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

fix update preview samples after disable pretrain

Browse source
This commit is contained in:
Colombo 2020-01-28 13:32:01 +04:00
parent 7386a9d6fd
commit 9c6ca24642
3 changed files with 61 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/Model_Quick96/Model.py
View file

@ -337,11 +337,12 @@ class QModel(ModelBase):
# Loading/initializing all models/optimizers weights
for model, filename in io.progress_bar_generator(self.model_filename_list, "Initializing models"):
do_init = self.is_first_run()
if self.pretrain_just_disabled:
do_init = False
if model == self.inter:
do_init = True
else:
do_init = self.is_first_run()
if not do_init:
do_init = not model.load_weights( self.get_strpath_storage_for_file(filename) )