From 4b8d8cdca11c17a1a54504045c56fc8cc6da06fc Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Fri, 13 Mar 2020 14:24:21 +0100
Subject: [PATCH] added check, if lr_decay is set
---
src/run_modules/model_setup.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py
index 307fd630..54d150e0 100644
--- a/src/run_modules/model_setup.py
+++ b/src/run_modules/model_setup.py
@@ -70,11 +70,12 @@ class ModelSetup(RunEnvironment):
Set all callbacks for the training phase. Add all callbacks with the .add_callback statement. Finally, the
advanced model checkpoint is added.
"""
- lr = self.data_store.get("lr_decay", scope="general.model")
+ lr = self.data_store.get_default("lr_decay", scope="general.model", default=None)
hist = HistoryAdvanced()
self.data_store.set("hist", hist, scope="general.model")
callbacks = CallbackHandler()
- callbacks.add_callback(lr, self.callbacks_name % "lr", "lr")
+ if lr:
+ callbacks.add_callback(lr, self.callbacks_name % "lr", "lr")
callbacks.add_callback(hist, self.callbacks_name % "hist", "hist")
callbacks.create_model_checkpoint(filepath=self.checkpoint_name, verbose=1, monitor='val_loss',
save_best_only=True, mode='auto')
--
GitLab