david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Colombo
2020-03-04 17:54:52 +04:00
parent cd76425db3
commit 9365e42f25
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mainscripts/Trainer.py
+3 -5
View File
@@ -33,15 +33,13 @@ def trainerThread (s2c, c2s, e,
save_interval_min = 15 save_interval_min = 15
if not training_data_src_path.exists(): if not training_data_src_path.exists():
io.log_err('Training data src directory does not exist.') training_data_src_path.mkdir(exist_ok=True, parents=True)
break
if not training_data_dst_path.exists(): if not training_data_dst_path.exists():
io.log_err('Training data dst directory does not exist.') training_data_dst_path.mkdir(exist_ok=True, parents=True)
break
if not saved_models_path.exists(): 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)( model = models.import_model(model_class_name)(
is_training=True, is_training=True,