Compare commits

...

1 Commits

Author SHA1 Message Date
Jonathan Raiman
a4fba209c4 remove ref to simple_bench + rank variable not used [mpi gone] 2017-10-25 14:08:01 -07:00
2 changed files with 1 additions and 2 deletions

View File

@@ -13,7 +13,7 @@ def train(env_id, num_timesteps, seed, num_cpu):
def _thunk():
env = make_atari(env_id)
env.seed(seed + rank)
env = bench.Monitor(env, logger.get_dir() and os.path.join(logger.get_dir(), str(rank)))
env = bench.Monitor(env, logger.get_dir() and logger.get_dir())
gym.logger.setLevel(logging.WARN)
return wrap_deepmind(env)
return _thunk

View File

@@ -1,3 +1,2 @@
from baselines.bench.benchmarks import *
from baselines.bench.monitor import *
from baselines.bench.simple_bench import simple_bench