Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 31, 2021

Commit

51b4b2d

1 Parent(s): 26689f9

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -37,7 +37,7 @@ def train_model():
     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:

     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: