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

Set 'cnn' as default network for retro (#683)

Browse Source
This commit is contained in:
Mathieu Poliquin
2018-10-30 04:30:41 +08:00
committed by pzhokhov
parent 8e56ddeac2
commit e2b41828af
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

@@ -131,7 +131,7 @@ def get_env_type(env_id):
def get_default_network(env_type): def get_default_network(env_type):
if env_type == 'atari': if env_type in {'atari', 'retro'}:
return 'cnn' return 'cnn'
else: else:
return 'mlp' return 'mlp'