This commit is contained in:
Alexander Munch-Hansen 2018-04-24 22:31:18 +02:00
commit ea3f05846d

View File

@ -34,6 +34,10 @@ parser.add_argument('--list-models', action='store_true',
args = parser.parse_args() args = parser.parse_args()
if args.model == "baseline":
print("Model name 'baseline' not allowed")
exit()
config = { config = {
'model': args.model, 'model': args.model,
'episode_count': args.episode_count, 'episode_count': args.episode_count,