Gagan Bhatia commited on
Commit
2a27f03
1 Parent(s): 6445c8d

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +0 -1
src/models/model.py CHANGED
@@ -313,7 +313,6 @@ class Summarization:
313
  tokenizer=self.tokenizer, model=self.model, output=outputdir
314
  )
315
 
316
- # monitor="train_loss",
317
  # mode="min",
318
  # )
319
  MLlogger = MLFlowLogger(experiment_name="Summarization",
 
313
  tokenizer=self.tokenizer, model=self.model, output=outputdir
314
  )
315
 
 
316
  # mode="min",
317
  # )
318
  MLlogger = MLFlowLogger(experiment_name="Summarization",