Skip to content

Commit 0840a82

Browse files
fix bug in SAC
1 parent 0394b6b commit 0840a82

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

rlzoo/algorithms/sac/run_sac.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,9 @@
3232
hidden_dim_list=num_hidden_layer * [hidden_dim])
3333
with tf.name_scope('Policy'):
3434
policy_net = StochasticPolicyNetwork(env.observation_space, env.action_space,
35-
hidden_dim_list=num_hidden_layer * [hidden_dim], state_conditioned=True)
35+
hidden_dim_list=num_hidden_layer * [hidden_dim],
36+
output_activation=None,
37+
state_conditioned=True)
3638
net_list = [soft_q_net1, soft_q_net2, target_soft_q_net1, target_soft_q_net2, policy_net]
3739

3840
""" choose optimizers """

0 commit comments

Comments
 (0)