Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 31, 2021

Commit

26689f9

1 Parent(s): b2f69cf

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -38,7 +38,6 @@ def train_model():
     model.save_model(model_dir=params["model_dir"])
     with open('wandb/latest-run/files/wandb-summary.json') as json_file:
-        model_type=params["model_type"], model_name=params["model_name"]
         data = json.load(json_file)
     with open("reports/training_metrics.txt", "w") as fp:

     model.save_model(model_dir=params["model_dir"])
     with open('wandb/latest-run/files/wandb-summary.json') as json_file:
         data = json.load(json_file)
     with open("reports/training_metrics.txt", "w") as fp: