Gagan Bhatia commited on
Commit
d5e186f
·
1 Parent(s): 1e7e922

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -290,6 +290,7 @@ class Summarization:
290
  early_stopping_patience_epochs: int = 0, # 0 to disable early stopping feature
291
  learning_rate: float = 0.0001,
292
  adam_epsilon: float = 0.01,
 
293
  ):
294
  """
295
  trains T5/MT5 model on custom dataset
 
290
  early_stopping_patience_epochs: int = 0, # 0 to disable early stopping feature
291
  learning_rate: float = 0.0001,
292
  adam_epsilon: float = 0.01,
293
+ num_workers: int = 2,
294
  ):
295
  """
296
  trains T5/MT5 model on custom dataset