add save_config

king-menin · king-menin · commit 779d8d494faa · 2019-01-19T18:12:53.000+03:00
diff --git a/modules/train/train.py b/modules/train/train.py
@@ -190,6 +190,10 @@ def __init__(self, model, data, best_model_path, lr=0.001, betas=[0.8, 0.9], cli
         self.best_target_metric = 0.
         self.lr_scheduler = None
 
+    def save_config(self, path):
+        with open(path, "w") as file:
+            json.dump(self.config, file)
+
     @classmethod
     def from_config(cls, path):
         with open(path, "r") as file:
