subproc env

agi-brain · Nov 2, 2023 · c9a644a · c9a644a
1 parent b42bb20
commit c9a644a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/xuance/configs/a2c/classic_control/CartPole-v1.yaml b/xuance/configs/a2c/classic_control/CartPole-v1.yaml
@@ -1,7 +1,7 @@
 agent: "A2C"
 env_name: "Classic Control"
 env_id: "CartPole-v1"
-vectorize: "Dummy_Gym"
+vectorize: "Subproc_Gym"
 policy: "Categorical_AC"
 representation: "Basic_MLP"
 runner: "DRL"

diff --git a/xuance/environment/__init__.py b/xuance/environment/__init__.py
@@ -22,7 +22,7 @@
     "Dummy_Atari": DummyVecEnv_Atari,
 
     # multiprocess #
-    "Subproc": SubprocVecEnv,
+    # "Subproc": SubprocVecEnv,
     "Subproc_Gym": SubprocVecEnv_Gym,
     "Subproc_Pettingzoo": SubprocVecEnv_Pettingzoo,
     "Subproc_StarCraft2": SubprocVecEnv_StarCraft2,

diff --git a/xuance/torch/runners/runner_sc2.py b/xuance/torch/runners/runner_sc2.py
@@ -292,7 +292,7 @@ def benchmark(self):
             "step": self.current_step
         }
 
-        agent_info = f"({self.args.agent}, seed={self.args.seed})"
+        agent_info = f"({self.args.agent}, {self.args.env_id}, seed={self.args.seed})"
         time_start = time.time()
         while self.current_step <= self.running_steps:
             print(agent_info, f"Steps: {self.current_step} / {self.running_steps}: ")