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

This commit is contained in:
Peter Zhokhov
2018-10-29 15:25:38 -07:00

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'