From 67df72fde282adb6dd6e0ef39b8274de09814541 Mon Sep 17 00:00:00 2001 From: Buridi Aditya Date: Sun, 28 Oct 2018 19:27:01 +0530 Subject: [PATCH] Supports for Madras Env --- baselines/ddpg/ddpg.py | 2 +- baselines/run.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/baselines/ddpg/ddpg.py b/baselines/ddpg/ddpg.py index 8b8659bee6..3b1a65d923 100755 --- a/baselines/ddpg/ddpg.py +++ b/baselines/ddpg/ddpg.py @@ -51,7 +51,7 @@ def learn(network, env, rank = MPI.COMM_WORLD.Get_rank() nb_actions = env.action_space.shape[-1] - assert (np.abs(env.action_space.low) == env.action_space.high).all() # we assume symmetric actions. + #assert (np.abs(env.action_space.low) == env.action_space.high).all() # we assume symmetric actions. memory = Memory(limit=int(1e6), action_shape=env.action_space.shape, observation_shape=env.observation_space.shape) critic = Critic(network=network, **network_kwargs) diff --git a/baselines/run.py b/baselines/run.py index 4aaf1a7010..5753f30eec 100644 --- a/baselines/run.py +++ b/baselines/run.py @@ -49,7 +49,7 @@ 'SpaceInvaders-Snes', } - +_game_envs['madras'] = {'gym-torcs-v0','gym-madras-v0'} def train(args, extra_args): env_type, env_id = get_env_type(args.env) print('env_type: {}'.format(env_type))