From 2e62741ac8f34e29109e830c1f9e98f209c7142e Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Wed, 12 Feb 2020 13:18:11 +0100
Subject: [PATCH] just train if trainable is true, otherwise rise info in
logger
---
src/run_modules/training.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/run_modules/training.py b/src/run_modules/training.py
index 7eb1cd7a..ff2cffcd 100644
--- a/src/run_modules/training.py
+++ b/src/run_modules/training.py
@@ -28,6 +28,7 @@ class Training(RunEnvironment):
self.lr_sc = self.data_store.get("lr_decay", "general.model")
self.hist = self.data_store.get("hist", "general.model")
self.experiment_name = self.data_store.get("experiment_name", "general")
+ self._trainable = self.data_store.get("trainable", "general")
self._run()
def _run(self) -> None:
@@ -44,8 +45,11 @@ class Training(RunEnvironment):
"""
self.set_generators()
self.make_predict_function()
- self.train()
- self.save_model()
+ if self._trainable:
+ self.train()
+ self.save_model()
+ else:
+ logging.info("No training has started, because trainable parameter was false.")
def make_predict_function(self) -> None:
"""
--
GitLab