From 42ecd1423bb35fa810c89e894c5dc26c6d5b5510 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Thu, 12 Dec 2019 14:39:24 +0100
Subject: [PATCH] save best model after training in data store
---
src/run_modules/training.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/src/run_modules/training.py b/src/run_modules/training.py
index 5dab3da2..53a50548 100644
--- a/src/run_modules/training.py
+++ b/src/run_modules/training.py
@@ -93,6 +93,7 @@ class Training(RunEnvironment):
model_name = os.path.join(path, name)
logging.debug(f"save best model to {model_name}")
self.model.save(model_name)
+ self.data_store.put("best_model", self.model, "general")
def load_best_model(self, name: str) -> None:
"""
@@ -119,6 +120,3 @@ class Training(RunEnvironment):
json.dump(history.history, f)
with open(os.path.join(path, "history_lr.json"), "w") as f:
json.dump(self.lr_sc.lr, f)
-
-
-
--
GitLab