From e4b2965b45c560cb7ba67eeee4c570f8899228df Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Sat, 8 Oct 2022 10:12:13 +0530
Subject: [PATCH] log config

---
 enhancer/cli/train.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py
index a9c66e0..81fd443 100644
--- a/enhancer/cli/train.py
+++ b/enhancer/cli/train.py
@@ -3,7 +3,7 @@ from types import MethodType
 
 import hydra
 from hydra.utils import instantiate
-from omegaconf import DictConfig
+from omegaconf import DictConfig, OmegaConf
 from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
 from pytorch_lightning.loggers import MLFlowLogger
 from torch.optim.lr_scheduler import ReduceLROnPlateau
@@ -15,12 +15,16 @@ JOB_ID = os.environ.get("SLURM_JOBID", "0")
 @hydra.main(config_path="train_config", config_name="config")
 def main(config: DictConfig):
 
+    yaml_conf = OmegaConf.to_yaml(config)
+    OmegaConf.save(yaml_conf, "config_log.yaml")
+
     callbacks = []
     logger = MLFlowLogger(
         experiment_name=config.mlflow.experiment_name,
         run_name=config.mlflow.run_name,
         tags={"JOB_ID": JOB_ID},
     )
+    logger.experiment.log_artifact(logger.run_id, "config_log.yaml")
 
     parameters = config.hyperparameters