Gagan Bhatia commited on
Commit
72de944
1 Parent(s): c9bef48

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -316,6 +316,7 @@ class Summarization:
316
  MLlogger = MLFlowLogger(experiment_name="Summarization",
317
  tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
318
 
 
319
 
320
  early_stop_callback = (
321
  [
 
316
  MLlogger = MLFlowLogger(experiment_name="Summarization",
317
  tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
318
 
319
+ logger = DAGsHubLogger(metrics_path='reports/metrics.txt')
320
 
321
  early_stop_callback = (
322
  [