Bug fix for espnet#76

Masao-Someki · Oct 14, 2023 · 729597c · 729597c
1 parent 318eacd
commit 729597c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espnet_onnx/export/asr/models/encoders/contextual_block_xformer.py b/espnet_onnx/export/asr/models/encoders/contextual_block_xformer.py
@@ -104,7 +104,7 @@ def forward(
         ]  # (B, L, overlap)
 
         block_num = max(0, xs_pad.size(1) - self.overlap_size) // self.hop_size
-        res_frame_num = xs_pad.size(1) - self.hop_size * block_num - 1
+        res_frame_num = xs_pad.size(1) - self.hop_size * block_num
         buffer_after_downsampling = xs_pad[:, -res_frame_num:]
         xs_pad = xs_pad[:, : block_num * self.hop_size + self.overlap_size]