debug

Denys88 · May 23, 2022 · 7a49df2 · 7a49df2
1 parent 1795f39
commit 7a49df2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/rl_games/common/a2c_common.py b/rl_games/common/a2c_common.py
@@ -361,6 +361,7 @@ def get_action_values(self, obs):
         return res_dict
 
     def get_values(self, obs):
+        print(f"self.local_rank = {self.local_rank}, obs.sum() = {obs['obs'].sum()}")
         with torch.no_grad():
             if self.has_central_value:
                 states = obs['states']

diff --git a/rl_games/torch_runner.py b/rl_games/torch_runner.py
@@ -61,7 +61,8 @@ def load_config(self, params):
         self.exp_config = None
 
         if self.seed:
-
+
+            print(f"====================================, self.rank = {self.rank}, self.seed = {self.seed}")
             torch.manual_seed(self.seed)
             torch.cuda.manual_seed_all(self.seed)
             np.random.seed(self.seed)