From d7f2d79ee7bba8afdb31179a8ac8db17f232e371 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Thu, 4 Nov 2021 12:18:10 +0100
Subject: [PATCH] fix issue if checkpoint has no model
---
mlair/model_modules/keras_extensions.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/model_modules/keras_extensions.py b/mlair/model_modules/keras_extensions.py
index a5fdcea7..8b99acd0 100644
--- a/mlair/model_modules/keras_extensions.py
+++ b/mlair/model_modules/keras_extensions.py
@@ -353,7 +353,7 @@ class CallbackHandler:
for pos, callback in enumerate(self.__callbacks):
path = callback["path"]
clb = pickle.load(open(path, "rb"))
- if clb.model is None:
+ if clb.model is None and hasattr(self._checkpoint, "model"):
clb.model = self._checkpoint.model
self._update_callback(pos, clb)
--
GitLab