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