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

minor change to install.py and baselines run.py (#121)

Browse Source
This commit is contained in:
Christopher Hesse
2018-10-01 10:38:07 -07:00
committed by Peter Zhokhov
parent 4cf7dc9644
commit 9ee804c384
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
baselines/run.py
View File

@@ -113,7 +113,7 @@ def build_env(args):
elif env_type == 'retro': elif env_type == 'retro':
import retro import retro
gamestate = args.gamestate or 'Level1-1' gamestate = args.gamestate or retro.State.DEFAULT
env = retro_wrappers.make_retro(game=args.env, state=gamestate, max_episode_steps=10000, env = retro_wrappers.make_retro(game=args.env, state=gamestate, max_episode_steps=10000,
use_restricted_actions=retro.Actions.DISCRETE) use_restricted_actions=retro.Actions.DISCRETE)
env.seed(args.seed) env.seed(args.seed)