diff --git a/rl_games/common/a2c_common.py b/rl_games/common/a2c_common.py index 9b47af56..e93c5f15 100644 --- 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 index e697b1ed..eb02db96 100644 --- 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)