Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 31, 2021

Commit

226f83d

1 Parent(s): e3ba5c5

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"]
     )
-                batch_size=params['batch_size'], max_epochs=params['epochs'],
                 use_gpu=params['use_gpu'], learning_rate=float(params['learning_rate']),
                 num_workers=int(params['num_workers']))

         model_type=params["model_type"], model_name=params["model_name"]
     )
                 use_gpu=params['use_gpu'], learning_rate=float(params['learning_rate']),
                 num_workers=int(params['num_workers']))