From 33a4b0db3c75bfc7ad2b4020997a850afb29ac4d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Christoffer=20M=C3=BCller=20Madsen?= Date: Tue, 24 Apr 2018 21:16:54 +0200 Subject: [PATCH] disallow using model "baseline" --- main.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/main.py b/main.py index f6a375e..513f076 100644 --- a/main.py +++ b/main.py @@ -34,6 +34,10 @@ parser.add_argument('--list-models', action='store_true', args = parser.parse_args() +if args.model == "baseline": + print("Model name 'baseline' not allowed") + exit() + config = { 'model': args.model, 'episode_count': args.episode_count,