Gagan Bhatia commited on
Commit
adf7210
1 Parent(s): 7f81350

Update model.py

Browse files
Files changed (1) hide show
  1. 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(