Skip to content
Snippets Groups Projects
Commit 2e62741a authored by lukas leufen's avatar lukas leufen
Browse files

just train if trainable is true, otherwise rise info in logger

parent 315995c2
Branches
No related tags found
2 merge requests!50release for v0.7.0,!38Lukas issue043 refac clearify use of trainable option
Pipeline #29405 passed
...@@ -28,6 +28,7 @@ class Training(RunEnvironment): ...@@ -28,6 +28,7 @@ class Training(RunEnvironment):
self.lr_sc = self.data_store.get("lr_decay", "general.model") self.lr_sc = self.data_store.get("lr_decay", "general.model")
self.hist = self.data_store.get("hist", "general.model") self.hist = self.data_store.get("hist", "general.model")
self.experiment_name = self.data_store.get("experiment_name", "general") self.experiment_name = self.data_store.get("experiment_name", "general")
self._trainable = self.data_store.get("trainable", "general")
self._run() self._run()
def _run(self) -> None: def _run(self) -> None:
...@@ -44,8 +45,11 @@ class Training(RunEnvironment): ...@@ -44,8 +45,11 @@ class Training(RunEnvironment):
""" """
self.set_generators() self.set_generators()
self.make_predict_function() self.make_predict_function()
if self._trainable:
self.train() self.train()
self.save_model() self.save_model()
else:
logging.info("No training has started, because trainable parameter was false.")
def make_predict_function(self) -> None: def make_predict_function(self) -> None:
""" """
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment