Player games num fix.

Denys88 · Oct 1, 2024 · 54d5eb6 · 54d5eb6
1 parent 9bb9862
commit 54d5eb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rl_games/common/player.py b/rl_games/common/player.py
@@ -62,7 +62,7 @@ def __init__(self, params):
             'central_value_config') is not None
         self.device_name = self.config.get('device_name', 'cuda')
         self.render_env = self.player_config.get('render', False)
-        self.games_num = self.player_config.get('games_num', 2000)
+        self.games_num = self.player_config.get('games_num', 1000000000)
 
         if 'deterministic' in self.player_config:
             self.is_deterministic = self.player_config['deterministic']
@@ -73,7 +73,7 @@ def __init__(self, params):
         self.n_game_life = self.player_config.get('n_game_life', 1)
         self.print_stats = self.player_config.get('print_stats', True)
         self.render_sleep = self.player_config.get('render_sleep', 0.002)
-        self.max_steps = 108000 // 4
+        self.max_steps = self.player_config.get('max_steps', 100000000)
         self.device = torch.device(self.device_name)
 
         self.evaluation = self.player_config.get("evaluation", False)