Update cif_loss.py

Mddct · Mar 11, 2023 · df0e2ae · df0e2ae
1 parent 55c9323
commit df0e2ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/cif_loss.py b/py/cif_loss.py
@@ -8,7 +8,7 @@ def __init__(self,
                  spike_threshold: float = 0.0,
                  blank_id: int = 0) -> None:
         super().__init__()
-        self.spike_threshold = math.log(spike_threshold)
+        self.spike_threshold = math.log(1-spike_threshold)
         self.blank = blank_id
 
     def forward(self, alpha: torch.Tensor, ctc_log_probs: torch.Tensor,
@@ -19,7 +19,7 @@ def forward(self, alpha: torch.Tensor, ctc_log_probs: torch.Tensor,
         text_mask = make_non_pad_mask(text_length)
         batch_size = alpha.size(0)
         ctc_blank_probs = ctc_log_probs[:, :, self.blank]
-        triggerd = (math.log(1) - ctc_blank_probs) > self.spike_threshold
+        triggerd = ctc_blank_probs < self.spike_threshold
         spikes = triggerd * mask
         begin = torch.ones(batch_size,
                            1,