github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-30 11:40:36 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

8
mainscripts/Trainer.py
View file

@ -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,