add "--force-creation" flag to force model creation

main.py

@@ -31,11 +31,13 @@ parser.add_argument('--train-perpetually', action='store_true',
                     help='start new training session as soon as the previous is finished')
 parser.add_argument('--list-models', action='store_true',
                     help='list all known models')
+parser.add_argument('--force-creation', action='store_true',
+                    help='force model creation if model does not exist')
 
 args = parser.parse_args()
 
-if args.model == "baseline":
-    print("Model name 'baseline' not allowed")
+if args.model == "baseline_model":
+    print("Model name 'baseline_model' not allowed")
     exit()
 
 config = {
@@ -52,6 +54,7 @@ config = {
     'model_storage_path': 'models',
     'bench_storage_path': 'bench',
     'board_representation': 'quack'
+    'force_creation': args.force_creation
 }
 
 # Create models folder
@@ -68,7 +71,6 @@ if not os.path.isdir(log_path):
     os.mkdir(log_path)
 
 
-
 # Define helper functions
 def log_train_outcome(outcome, trained_eps = 0, log_path = os.path.join(model_path(), 'logs', "train.log")):
     format_vars = { 'trained_eps': trained_eps,

network.py

@@ -152,7 +152,7 @@ class Network:
                 print("Variable: ", k)
                 print("Shape: ", v.shape)
                 print(v)
-        else:
+        elif not self.config['force_creation']:
             print("You need to have baseline_model inside models")
             exit()