silly bug

montrealrobotics · Oct 6, 2023 · 6b49954 · 6b49954
1 parent e273250
commit 6b49954
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanrl/ppo_continuous_action_wandb.py b/cleanrl/ppo_continuous_action_wandb.py
@@ -682,7 +682,7 @@ def train(cfg):
                 #print(global_step)
                 # update_risk = 0
                 # while update_risk < cfg.num_update_risk:
-                for epoch in cfg.num_risk_epochs:
+                for epoch in range(cfg.num_risk_epochs):
                     if cfg.finetune_risk_online:
                         print("I am online")
                         data = rb.slice_data(-cfg.risk_batch_size*cfg.num_update_risk, 0)