nothing interesting

2019-01-24 21:28:54 +04:00
parent 403343e465
commit 1e7a0836f7
1 changed files with 5 additions and 0 deletions
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -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' )