Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
adf7210
1
Parent(s): 7f81350
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
|
@@ -302,6 +302,7 @@ class Summarization:
|
|
| 302 |
if val_loss does not improve after the specied number of epochs. set 0 to disable early stopping.
|
| 303 |
Defaults to 0 (disabled)
|
| 304 |
:param learning_rate:
|
|
|
|
| 305 |
"""
|
| 306 |
self.target_max_token_len = target_max_token_len
|
| 307 |
self.data_module = PLDataModule(
|
|
|
|
| 302 |
if val_loss does not improve after the specied number of epochs. set 0 to disable early stopping.
|
| 303 |
Defaults to 0 (disabled)
|
| 304 |
:param learning_rate:
|
| 305 |
+
:param adam_epsilon:
|
| 306 |
"""
|
| 307 |
self.target_max_token_len = target_max_token_len
|
| 308 |
self.data_module = PLDataModule(
|