Skip to content
Snippets Groups Projects
Commit d7f2d79e authored by leufen1's avatar leufen1
Browse files

fix issue if checkpoint has no model

parent a43bbba0
No related branches found
No related tags found
5 merge requests!413update release branch,!412Resolve "release v2.0.0",!361name of pdf starts now with feature_importance, there is now also another...,!350Resolve "upgrade code to TensorFlow V2",!335Resolve "upgrade code to TensorFlow V2"
Pipeline #82171 failed
...@@ -353,7 +353,7 @@ class CallbackHandler: ...@@ -353,7 +353,7 @@ class CallbackHandler:
for pos, callback in enumerate(self.__callbacks): for pos, callback in enumerate(self.__callbacks):
path = callback["path"] path = callback["path"]
clb = pickle.load(open(path, "rb")) 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 clb.model = self._checkpoint.model
self._update_callback(pos, clb) self._update_callback(pos, clb)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment