add A2C

codes/A2C/env.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+# coding=utf-8
+'''
+Author: John
+Email: johnjim0816@gmail.com
+Date: 2020-10-30 15:39:37
+LastEditor: John
+LastEditTime: 2020-11-03 20:52:07
+Discription: 
+Environment: 
+'''
+
+import gym
+from common.multiprocessing_env import SubprocVecEnv
+
+# num_envs = 16
+# env_name = "Pendulum-v0"
+
+def make_envs(num_envs=16,env_name="Pendulum-v0"):
+    ''' 创建多个子环境
+    '''
+    num_envs = 16
+    env_name = "CartPole-v0"
+    def make_env():
+        def _thunk():
+            env = gym.make(env_name)
+            return env
+
+        return _thunk
+
+    envs = [make_env() for i in range(num_envs)]
+    envs = SubprocVecEnv(envs)
+    return envs
+# if __name__ == "__main__":
+
+#     num_envs = 16
+#     env_name = "CartPole-v0"
+#     def make_env():
+#         def _thunk():
+#             env = gym.make(env_name)
+#             return env
+
+#         return _thunk
+
+#     envs = [make_env() for i in range(num_envs)]
+#     envs = SubprocVecEnv(envs)
+if __name__ == "__main__":
+    envs = make_envs(num_envs=16,env_name="CartPole-v0")