Spaces:

gagan3012
/

summarization

Runtime error

gagan3012 commited on Jul 28, 2021

Commit

b3f69b2

1 Parent(s): bd5eb97

fixes

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -328,7 +328,8 @@ class Summarization:
         )
         MLlogger = MLFlowLogger(experiment_name="Summarization",
-                               tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
         # WandLogger = WandbLogger(project="summarization-dagshub")

         )
         MLlogger = MLFlowLogger(experiment_name="Summarization",
+                               tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow",
+                                save_dir='reports/training_metrics.txt')
         # WandLogger = WandbLogger(project="summarization-dagshub")