first steps towards discrete action space

montrealrobotics · Sep 22, 2023 · 28e13f8 · 28e13f8
1 parent 96e0238
commit 28e13f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanrl/dqn.py b/cleanrl/dqn.py
@@ -140,7 +140,7 @@ def linear_schedule(start_e: float, end_e: float, duration: int, t: int):
         )
     args = parse_args()
     run_name = f"{args.env_id}__{args.exp_name}__{args.seed}__{int(time.time())}"
-    action_map = {0: np.array([args.action_scale, args.action_scale]), 1: np.array([0., args.action_scale]), 2: np.array([args.action_scale, 0.])}
+    action_map = {0: np.array([args.action_scale, args.action_scale]), 1: np.array([-args.action_scale, args.action_scale]), 2: np.array([args.action_scale, -args.action_scale]), 3: np.array([-args.action_scale, -args.action_scale])}
     def action_map_fn(action):
         return action_map[action]