Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
bcf1243
1
Parent(s): 5e3ef77
Update model.py
Browse files- src/models/model.py +4 -2
src/models/model.py
CHANGED
|
@@ -349,8 +349,10 @@ class Summarization:
|
|
| 349 |
weight_decay=weight_decay,
|
| 350 |
)
|
| 351 |
|
| 352 |
-
MLlogger = MLFlowLogger(
|
| 353 |
-
|
|
|
|
|
|
|
| 354 |
|
| 355 |
WandLogger = WandbLogger(project="summarization-dagshub")
|
| 356 |
|
|
|
|
| 349 |
weight_decay=weight_decay,
|
| 350 |
)
|
| 351 |
|
| 352 |
+
MLlogger = MLFlowLogger(
|
| 353 |
+
experiment_name="Summarization",
|
| 354 |
+
tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow",
|
| 355 |
+
)
|
| 356 |
|
| 357 |
WandLogger = WandbLogger(project="summarization-dagshub")
|
| 358 |
|