conformer icon indicating copy to clipboard operation
conformer copied to clipboard

About the "input length mismatch" bug in torchaudio's RNNT loss

Open Zain-Jiang opened this issue 3 years ago • 2 comments

In conformer/convolution.py, line 183, the code

output_lengths = input_lengths >> 2
output_lengths -= 1

when the result of input_lengths >>2 is xx.75, the torchaudio.transforms.RNNTLoss will raise "input length mismatch" Error. Maybe this is a bug when calculating the output lengths, I'm not sure of it.

Zain-Jiang avatar Nov 10 '21 08:11 Zain-Jiang

I've gotten same problem. Have you fixed that yet?

longlnOff avatar Nov 17 '23 02:11 longlnOff