Merge branch 'master' of github.com:openai/baselines
This commit is contained in:
@@ -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'
|
||||||
|
Reference in New Issue
Block a user