disallow using model "baseline"

This commit is contained in:
Christoffer Müller Madsen 2018-04-24 21:16:54 +02:00
parent 349ad718f1
commit 33a4b0db3c

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,