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

Update model.py

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