add: epoch info in each epoch json in history

kirianguiller · Apr 3, 2023 · 80918f1 · 80918f1
1 parent c9ba2a4
commit 80918f1
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/BertForDeprel/parser/cmds/train.py b/BertForDeprel/parser/cmds/train.py
@@ -170,10 +170,10 @@ def __call__(self, args, model_params: ModelParams_T):
         results = model.eval_epoch(test_loader, args.device)
         results["n_sentences_train"] = len(train_dataset)
         results["n_sentences_test"] = len(test_dataset)
+        results["epoch"] = n_epoch_start
 
-        history = {
-            n_epoch_start: results,
-        }
+        history = []
+        history.append(results)
         best_loss = results["loss_epoch"]
         best_LAS = results["LAS_epoch"]
         best_epoch_results = results
@@ -185,7 +185,9 @@ def __call__(self, args, model_params: ModelParams_T):
             results = model.eval_epoch(test_loader, args.device)
             results["n_sentences_train"] = len(train_dataset)
             results["n_sentences_test"] = len(test_dataset)
-            history[n_epoch] = results
+            results["epoch"] = n_epoch
+
+            history.append(results)
             with open(path_scores_history, "w") as outfile:
                 outfile.write(json.dumps(history))