From fd1226f7c6343d9b826b662b1f1134b0fad34cbb Mon Sep 17 00:00:00 2001 From: aegggsehyun1994 Date: Mon, 28 Dec 2020 23:00:56 +0900 Subject: [PATCH] Fixed configs to str --- configs/reacher_v2/ddpg.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/configs/reacher_v2/ddpg.py b/configs/reacher_v2/ddpg.py index d2a88f5c..d02523aa 100644 --- a/configs/reacher_v2/ddpg.py +++ b/configs/reacher_v2/ddpg.py @@ -3,9 +3,6 @@ - Author: Kyunghwan Kim - Contact: kh.kim@medipixel.io """ -import torch - -from rl_algorithms.common.helper_functions import identity agent = dict( type="DDPGAgent", @@ -25,12 +22,12 @@ head=dict( actor=dict( type="MLP", - configs=dict(hidden_sizes=[256, 256], output_activation=torch.tanh,), + configs=dict(hidden_sizes=[256, 256], output_activation="tanh",), ), critic=dict( type="MLP", configs=dict( - hidden_sizes=[256, 256], output_size=1, output_activation=identity, + hidden_sizes=[256, 256], output_size=1, output_activation="identity", ), ), ),