Gagan Bhatia commited on
Commit
0b69bd9
1 Parent(s): a88fd83

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -321,6 +321,7 @@ class Summarization:
321
  # monitor="train_loss",
322
  # mode="min",
323
  # )
 
324
 
325
  logger = MLFlowLogger(experiment_name="Summarization")
326
 
 
321
  # monitor="train_loss",
322
  # mode="min",
323
  # )
324
+ MLlogger = MLFlowLogger(experiment_name="Summarization",
325
 
326
  logger = MLFlowLogger(experiment_name="Summarization")
327