Gagan Bhatia commited on
Commit
bcf1243
1 Parent(s): 5e3ef77

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +4 -2
src/models/model.py CHANGED
@@ -349,8 +349,10 @@ class Summarization:
349
  weight_decay=weight_decay,
350
  )
351
 
352
- MLlogger = MLFlowLogger(experiment_name="Summarization",
353
- tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
 
 
354
 
355
  WandLogger = WandbLogger(project="summarization-dagshub")
356
 
 
349
  weight_decay=weight_decay,
350
  )
351
 
352
+ MLlogger = MLFlowLogger(
353
+ experiment_name="Summarization",
354
+ tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow",
355
+ )
356
 
357
  WandLogger = WandbLogger(project="summarization-dagshub")
358