From 9ee809a047ce7eb3b42c25da5a4adfc3c79b3858 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Tue, 15 Nov 2022 21:51:45 +0530
Subject: [PATCH] rename to train

---
 mayavoz/cli/train.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/mayavoz/cli/train.py b/mayavoz/cli/train.py
index c00c024..8f12ea7 100644
--- a/mayavoz/cli/train.py
+++ b/mayavoz/cli/train.py
@@ -19,9 +19,9 @@ JOB_ID = os.environ.get("SLURM_JOBID", "0")
 
 
 @hydra.main(config_path="train_config", config_name="config")
-def main(config: DictConfig):
+def train(config: DictConfig):
 
-    OmegaConf.save(config, "config_log.yaml")
+    OmegaConf.save(config, "config.yaml")
 
     callbacks = []
     logger = MLFlowLogger(
@@ -96,7 +96,7 @@ def main(config: DictConfig):
     trainer.test(model)
 
     logger.experiment.log_artifact(
-        logger.run_id, f"{trainer.default_root_dir}/config_log.yaml"
+        logger.run_id, f"{trainer.default_root_dir}/config.yaml"
     )
 
     saved_location = os.path.join(
@@ -117,4 +117,4 @@ def main(config: DictConfig):
 
 
 if __name__ == "__main__":
-    main()
+    train()