akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

bug fix [skip ci]

Browse source
This commit is contained in:
David Scripka 2023-11-21 17:09:11 -05:00
parent 08f31225a2
commit ee9270f93a
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
openwakeword/train.py
View file

@ -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]