fixes

params.yml

@@ -6,7 +6,7 @@ model_name: t5-small
 learning_rate: 1e-4
 epochs: 5
 source_dir: src
-model_dir: models/models/
+model_dir: models
 metric: rouge
 split: 0.02
 use_gpu: True

src/models/model.py

@@ -327,12 +327,12 @@ class Summarization:
             learning_rate=learning_rate, adam_epsilon=adam_epsilon, weight_decay=weight_decay
         )
 
-        # MLlogger = MLFlowLogger(experiment_name="Summarization",
-        #                        tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")\
+        MLlogger = MLFlowLogger(experiment_name="Summarization",
+                               tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
 
-        WandLogger = WandbLogger(project="summarization-dagshub")
+        # WandLogger = WandbLogger(project="summarization-dagshub")
 
-        # logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
+        #logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
 
         early_stop_callback = (
             [
@@ -351,7 +351,7 @@ class Summarization:
         gpus = -1 if use_gpu and torch.cuda.is_available() else 0
 
         trainer = Trainer(
-            logger=[WandLogger],
+            logger=[MLlogger],
             callbacks=early_stop_callback,
             max_epochs=max_epochs,
             gpus=gpus,
@@ -398,7 +398,7 @@ class Summarization:
 
     def save_model(
             self,
-            model_dir="../../models"
+            model_dir="models"
     ):
         """
         Save model to dir