Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
63f745e
1
Parent(s):
dde8aa5
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -330,6 +330,7 @@ class Summarization:
|
|
330 |
# MLlogger = MLFlowLogger(experiment_name="Summarization",
|
331 |
# tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")\
|
332 |
|
|
|
333 |
|
334 |
early_stop_callback = (
|
335 |
[
|
|
|
330 |
# MLlogger = MLFlowLogger(experiment_name="Summarization",
|
331 |
# tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")\
|
332 |
|
333 |
+
WandLogger = WandbLogger(project="summarization-dagshub")
|
334 |
|
335 |
early_stop_callback = (
|
336 |
[
|