bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
JohnJim0816
2021-04-16 14:59:23 +08:00
parent 312b57fdff
commit e4690ac89f
71 changed files with 805 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
codes/A2C/env.py
View File

@@ -14,16 +14,16 @@ import gym
from A2C.multiprocessing_env import SubprocVecEnv
# num_envs = 16
# env_name = "Pendulum-v0"
# env = "Pendulum-v0"
def make_envs(num_envs=16,env_name="Pendulum-v0"):
def make_envs(num_envs=16,env="Pendulum-v0"):
''' 创建多个子环境
'''
num_envs = 16
env_name = "CartPole-v0"
env = "CartPole-v0"
def make_env():
def _thunk():
env = gym.make(env_name)
env = gym.make(env)
return env
return _thunk
@@ -34,10 +34,10 @@ def make_envs(num_envs=16,env_name="Pendulum-v0"):
# if __name__ == "__main__":
# num_envs = 16
# env_name = "CartPole-v0"
# env = "CartPole-v0"
# def make_env():
# def _thunk():
# env = gym.make(env_name)
# env = gym.make(env)
# return env
# return _thunk
@@ -45,4 +45,4 @@ def make_envs(num_envs=16,env_name="Pendulum-v0"):
# 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")
envs = make_envs(num_envs=16,env="CartPole-v0")