From ee9270f93a0759fd9bb0d4298b3e108be4ef3a8a Mon Sep 17 00:00:00 2001 From: David Scripka Date: Tue, 21 Nov 2023 17:09:11 -0500 Subject: [PATCH] bug fix [skip ci] --- openwakeword/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/openwakeword/train.py b/openwakeword/train.py index 6f1ed63..7e468bf 100755 --- a/openwakeword/train.py +++ b/openwakeword/train.py @@ -561,7 +561,8 @@ class Model(nn.Module): 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_accuracy": self.history["val_accuracy"][-1]}) + "val_accuracy": self.history["val_accuracy"][-1], + "val_fp_per_hr": self.history.get("val_fp_per_hr", [0])[-1]}) self.best_val_recall = self.history["val_recall"][-1] self.best_val_accuracy = self.history["val_accuracy"][-1]