Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
7010421
1
Parent(s):
63f745e
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -332,6 +332,7 @@ class Summarization:
|
|
332 |
|
333 |
WandLogger = WandbLogger(project="summarization-dagshub")
|
334 |
|
|
|
335 |
early_stop_callback = (
|
336 |
[
|
337 |
EarlyStopping(
|
|
|
332 |
|
333 |
WandLogger = WandbLogger(project="summarization-dagshub")
|
334 |
|
335 |
+
# logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
|
336 |
early_stop_callback = (
|
337 |
[
|
338 |
EarlyStopping(
|