Gagan Bhatia commited on
Commit
25a6cb6
1 Parent(s): 61c646e

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -351,6 +351,7 @@ class Summarization:
351
  gpus = -1 if use_gpu and torch.cuda.is_available() else 0
352
 
353
  trainer = Trainer(
 
354
  callbacks=early_stop_callback,
355
  max_epochs=max_epochs,
356
  gpus=gpus,
 
351
  gpus = -1 if use_gpu and torch.cuda.is_available() else 0
352
 
353
  trainer = Trainer(
354
+ logger=[WandLogger, MLlogger],
355
  callbacks=early_stop_callback,
356
  max_epochs=max_epochs,
357
  gpus=gpus,