From 1e7a0836f71b24ee4540a3ef2ac4e89f0a163c9b Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 24 Jan 2019 21:28:54 +0400 Subject: [PATCH] nothing interesting --- models/ModelBase.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/models/ModelBase.py b/models/ModelBase.py index 69b73a4..7fc166f 100644 --- 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' )