Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
bb51a11
1
Parent(s):
19317c8
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
|
@@ -327,6 +327,7 @@ class Summarization:
|
|
| 327 |
learning_rate=learning_rate, adam_epsilon=adam_epsilon, weight_decay=weight_decay
|
| 328 |
)
|
| 329 |
|
|
|
|
| 330 |
|
| 331 |
WandLogger = WandbLogger(project="summarization-dagshub")
|
| 332 |
|
|
|
|
| 327 |
learning_rate=learning_rate, adam_epsilon=adam_epsilon, weight_decay=weight_decay
|
| 328 |
)
|
| 329 |
|
| 330 |
+
MLlogger = MLFlowLogger(experiment_name="Summarization",
|
| 331 |
|
| 332 |
WandLogger = WandbLogger(project="summarization-dagshub")
|
| 333 |
|