Gagan Bhatia commited on
Commit
ea40e7f
1 Parent(s): 7010421

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -333,6 +333,7 @@ class Summarization:
333
  WandLogger = WandbLogger(project="summarization-dagshub")
334
 
335
  # logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
 
336
  early_stop_callback = (
337
  [
338
  EarlyStopping(
 
333
  WandLogger = WandbLogger(project="summarization-dagshub")
334
 
335
  # logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
336
+
337
  early_stop_callback = (
338
  [
339
  EarlyStopping(