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

[Zain-Jiang]

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.

[longlnOff]

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

[zzzendurance]

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

I'm sorry to bother you. I would like to use the lengths of conformer, but I'm not sure what the lengths of input_lengths should be.

The following paragraph is a description of the parameters in the source code, does it mean that I input a one-dimensional tensor (that is, batch)? But then I saw in his example that the input was a three-dimensional tensor. How did you enter the parameters when you used the conformer?