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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -330,6 +330,7 @@ class Summarization:
330
  # MLlogger = MLFlowLogger(experiment_name="Summarization",
331
  # tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")\
332
 
 
333
 
334
  early_stop_callback = (
335
  [
 
330
  # MLlogger = MLFlowLogger(experiment_name="Summarization",
331
  # tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")\
332
 
333
+ WandLogger = WandbLogger(project="summarization-dagshub")
334
 
335
  early_stop_callback = (
336
  [