github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

Browse source
This commit is contained in:
iperov 2021-06-01 00:21:27 +04:00
parent 315f241c51
commit 1981ed0ca8
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
mainscripts/Trainer.py
View file

@ -43,7 +43,10 @@ def trainerThread (s2c, c2s, e,
if not saved_models_path.exists(): if not saved_models_path.exists():
saved_models_path.mkdir(exist_ok=True, parents=True) saved_models_path.mkdir(exist_ok=True, parents=True)
if dump_ckpt:
cpu_only=True
model = models.import_model(model_class_name)( model = models.import_model(model_class_name)(
is_training=not dump_ckpt, is_training=not dump_ckpt,
saved_models_path=saved_models_path, saved_models_path=saved_models_path,