Fix conditional statement

edbeeching · Dec 1, 2023 · 835a2e8 · 835a2e8
1 parent bebf23c
commit 835a2e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/stable_baselines3_example.py b/examples/stable_baselines3_example.py
@@ -151,7 +151,7 @@ def func(progress_remaining: float) -> float:
 
 
 if args.resume_model_path is None:
-    learning_rate = 0.0003 if args.inference else linear_schedule(0.0003)
+    learning_rate = 0.0003 if not args.linear_lr_schedule else linear_schedule(0.0003)
     model: PPO = PPO("MultiInputPolicy",
                      env,
                      ent_coef=0.0001,