From 06153869bbf8995e19af4b280af7278dccc16655 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Wed, 13 Jun 2018 19:28:43 +0200
Subject: [PATCH] Reload model weights if model checkpoint is used
---
toolkit.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/toolkit.py b/toolkit.py
index b1690df..7443c23 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -725,6 +725,9 @@ class ClassificationProject(object):
if not self.use_modelcheckpoint:
logger.info("Save weights")
self.model.save_weights(os.path.join(self.project_dir, "weights.h5"))
+ else:
+ self.model.load_weights(os.path.join(self.project_dir, "weights.h5"))
+ logger.info("Reloading weights, since we are using model checkpoint!")
self.total_epochs += epochs
self._write_info("epochs", self.total_epochs)
--
GitLab