disallow using model "baseline"

2018-04-24 21:16:54 +02:00 · 2018-04-24 21:16:54 +02:00 · 33a4b0db3c
commit 33a4b0db3c
parent 349ad718f1
1 changed files with 4 additions and 0 deletions
--- 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,