save best model after training in data store

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)
-
-
-