gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove ref to simple_bench + rank variable not used [mpi gone]

Browse Source
This commit is contained in:
Jonathan Raiman
2017-10-25 14:08:01 -07:00
parent bb40378118
commit a4fba209c4
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
baselines/acktr/run_atari.py
View File

@@ -13,7 +13,7 @@ def train(env_id, num_timesteps, seed, num_cpu):
def _thunk(): def _thunk():
env = make_atari(env_id) env = make_atari(env_id)
env.seed(seed + rank) 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) gym.logger.setLevel(logging.WARN)
return wrap_deepmind(env) return wrap_deepmind(env)
return _thunk return _thunk

1
baselines/bench/__init__.py
View File

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