Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 31, 2021

Commit

492b9c6

1 Parent(s): d8d799c

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -22,7 +22,6 @@ def train_model():
     model = Summarization()
-    model.train(train_df=train_df, eval_df=eval_df,
                 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 = Summarization()
                 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']))