quick fix

Denys88 · Nov 30, 2024 · 81f4570 · 81f4570
1 parent 6612eaf
commit 81f4570
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rl_games/common/a2c_common.py b/rl_games/common/a2c_common.py
@@ -386,7 +386,7 @@ def set_eval(self):
         if self.epochs_between_resets > 0:
             if self.epoch_num % self.epochs_between_resets == 0:
                 self.reset_envs()
-                self.init_current_rewards()
+                self.init_current_rewards(self.num_agents * self.num_actors, (self.num_agents * self.num_actors, self.value_size))
                 print(f"Forcing env reset after {self.epoch_num} epochs")