Gagan Bhatia commited on
Commit
8e49ded
1 Parent(s): e97a610

Update model.py

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