Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
7a797f0
1
Parent(s):
72de944
Update model.py
Browse files- 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,
|