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

nothing interesting

Browse source
This commit is contained in:
iperov 2019-01-24 21:28:54 +04:00
parent 403343e465
commit 1e7a0836f7
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/ModelBase.py
View file

@ -277,6 +277,11 @@ class ModelBase(object):
}
self.model_data_path.write_bytes( pickle.dumps(model_data) )
def load_weights_safe(self, model_filename_list):
for model, filename in model_filename_list:
if Path(filename).exists():
model.load_weights(filename)
def save_weights_safe(self, model_filename_list):
for model, filename in model_filename_list:
model.save_weights( filename + '.tmp' )