trainer now creates directories instead of throwing error when they do not exist

2020-03-04 17:54:52 +04:00
parent cd76425db3
commit 9365e42f25
1 changed files with 3 additions and 5 deletions
@@ -33,15 +33,13 @@ def trainerThread (s2c, c2s, e,
            save_interval_min = 15

            if not training_data_src_path.exists():
-                io.log_err('Training data src directory does not exist.')
-                break
+                training_data_src_path.mkdir(exist_ok=True, parents=True)

            if not training_data_dst_path.exists():
-                io.log_err('Training data dst directory does not exist.')
-                break
+                training_data_dst_path.mkdir(exist_ok=True, parents=True)

            if not saved_models_path.exists():
-                saved_models_path.mkdir(exist_ok=True)
+                saved_models_path.mkdir(exist_ok=True, parents=True)

            model = models.import_model(model_class_name)(
                        is_training=True,