Update cif_loss.py

Mddct · Mar 12, 2023 · cc4b37a · cc4b37a
1 parent c76e22a
commit cc4b37a
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/py/cif_loss.py b/py/cif_loss.py
@@ -21,11 +21,6 @@ def forward(self, alpha: torch.Tensor, ctc_log_probs: torch.Tensor,
         ctc_blank_probs = ctc_log_probs[:, :, self.blank]
         triggerd = ctc_blank_probs < self.spike_threshold
         spikes = triggerd * mask
-        begin = torch.ones(batch_size,
-                           1,
-                           dtype=spikes.dtype,
-                           device=spikes.device)
-        spikes = torch.cat([begin, spikes[:,1:]], dim=1)
         index = torch.arange(alpha.size(1),
                              device=alpha.device).unsqueeze(0)  #[1,L]
         boundary_loss_list = []