From c3bc8d6170aaa4a135fbdc2a1436bf5542209fad Mon Sep 17 00:00:00 2001 From: David Scripka Date: Tue, 21 Nov 2023 16:35:10 -0500 Subject: [PATCH] bug fix in checkpoint metric saving [skip ci] --- openwakeword/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/openwakeword/train.py b/openwakeword/train.py index 55595a9..465bdb5 100755 --- a/openwakeword/train.py +++ b/openwakeword/train.py @@ -560,7 +560,8 @@ class Model(nn.Module): logging.info("Saving checkpoint with metrics >= to targets!") self.best_models.append(copy.deepcopy(self.model)) self.best_model_scores.append({"training_step_ndx": step_ndx, "val_n_fp": self.history["val_n_fp"][-1], - "val_recall": self.history["val_recall"][-1]}) + "val_recall": self.history["val_recall"][-1]}, + "val_accuracy": self.history["val_accuracy"][-1]) self.best_val_recall = self.history["val_recall"][-1] self.best_val_accuracy = self.history["val_accuracy"][-1]