Skip to content

Commit 25dfc0a

Browse files
authored
bugfix for envs with max_action < 1
1 parent 8cb9f14 commit 25dfc0a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def evaluate_policy(policy, eval_episodes=10):
6767

6868
state_dim = env.observation_space.shape[0]
6969
action_dim = env.action_space.shape[0]
70-
max_action = int(env.action_space.high[0])
70+
max_action = float(env.action_space.high[0])
7171

7272
# Initialize policy
7373
if args.policy_name == "TD3": policy = TD3.TD3(state_dim, action_dim, max_action)

0 commit comments

Comments
 (0)