print config and args before training

luoming3 · Mar 6, 2024 · cb5dde7 · cb5dde7
1 parent 86b8c6f
commit cb5dde7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/offpolicy/scripts/train/train_custom.py b/offpolicy/scripts/train/train_custom.py
@@ -11,6 +11,8 @@
 from offpolicy.envs.custom.env_discrete import DiscreteActionEnv
 from offpolicy.envs.env_wrappers import DummyVecEnv, SubprocVecEnv
 
+import pprint
+
 
 def make_train_env(all_args):
     def get_env_fn(rank):
@@ -161,8 +163,8 @@ def policy_mapping_fn(agent_id): return 'policy_' + str(agent_id)
               "run_dir": run_dir
               }
 
-    import pprint
     pprint.pprint(config)
+    pprint.pprint(vars(all_args))
 
     total_num_steps = 0
     runner = Runner(config=config)