Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
8e49ded
1
Parent(s):
e97a610
Update model.py
Browse files- src/models/model.py +2 -0
src/models/model.py
CHANGED
@@ -283,6 +283,8 @@ class Summarization:
|
|
283 |
outputdir: str = "models",
|
284 |
early_stopping_patience_epochs: int = 0, # 0 to disable early stopping feature
|
285 |
learning_rate: float = 0.0001,
|
|
|
|
|
286 |
):
|
287 |
"""
|
288 |
trains T5/MT5 model on custom dataset
|
|
|
283 |
outputdir: str = "models",
|
284 |
early_stopping_patience_epochs: int = 0, # 0 to disable early stopping feature
|
285 |
learning_rate: float = 0.0001,
|
286 |
+
adam_epsilon: float = 0.01,
|
287 |
+
num_workers: int = 2
|
288 |
):
|
289 |
"""
|
290 |
trains T5/MT5 model on custom dataset
|