CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
models/ModelBase.py
+5
View File
@@ -277,6 +277,11 @@ class ModelBase(object):
} }
self.model_data_path.write_bytes( pickle.dumps(model_data) ) 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): def save_weights_safe(self, model_filename_list):
for model, filename in model_filename_list: for model, filename in model_filename_list:
model.save_weights( filename + '.tmp' ) model.save_weights( filename + '.tmp' )