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