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", ), ), ),