From 1eda27647b21fa6dec4ec632d780deebbabe74d1 Mon Sep 17 00:00:00 2001
From: Menglong Xu <32296227+mlxu995@users.noreply.github.com>
Date: Wed, 8 Dec 2021 13:41:20 +0800
Subject: [PATCH] [fix bug] add sigmoid() in score.py (#37)

The sigmoid() in kws/model/kws_model.py:KWSModel() was moved into kws/model/loss.py:max_pooling_loss()
To compute the posterior score correctly, the sigmoid() should also be added to kws/bin/score.py:main()
---
 kws/bin/score.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kws/bin/score.py b/kws/bin/score.py
index 4c80816..40986d2 100644
--- a/kws/bin/score.py
+++ b/kws/bin/score.py
@@ -102,7 +102,7 @@ def main():
             feats = feats.to(device)
             lengths = lengths.to(device)
             mask = padding_mask(lengths).unsqueeze(2)
-            logits = model(feats)
+            logits = torch.sigmoid(model(feats))
             logits = logits.masked_fill(mask, 0.0)
             max_logits, _ = logits.max(dim=1)
             max_logits = max_logits.cpu()