Merge branch 'master' of github.com:openai/baselines

2018-10-29 15:25:38 -07:00
parent de36116e3b e2b41828af
commit 93c7cc202c
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'