Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 31, 2021

Commit

9a78cd2

•

1 Parent(s): cf780e0

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -25,7 +25,6 @@ def train_model():
         model_type=params["model_type"], model_name=params["model_name"]
     )
-                num_workers=int(params['num_workers']))
     model.save_model(model_dir=params['model_dir'])

         model_type=params["model_type"], model_name=params["model_name"]
     )
     model.save_model(model_dir=params['model_dir'])