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 |
|