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

2018-10-30 04:30:41 +08:00
parent 8e56ddeac2
commit e2b41828af
1 changed files with 1 additions and 1 deletions
--- a/baselines/run.py
+++ b/baselines/run.py
@@ -131,7 +131,7 @@ def get_env_type(env_id):


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