Gagan Bhatia commited on
Commit
7a797f0
1 Parent(s): 72de944

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -336,6 +336,7 @@ class Summarization:
336
 
337
  trainer = Trainer(
338
  logger=logger,
 
339
  callbacks=early_stop_callback,
340
  max_epochs=max_epochs,
341
  gpus=gpus,
 
336
 
337
  trainer = Trainer(
338
  logger=logger,
339
+ logger=[logger, MLlogger],
340
  callbacks=early_stop_callback,
341
  max_epochs=max_epochs,
342
  gpus=gpus,