Conformer icon indicating copy to clipboard operation
Conformer copied to clipboard

Varying input shape

Open anasvaf opened this issue 2 years ago • 0 comments

Hello,

Thank you for translating the PyTorch script to TensorFlow :) I have a question about the input shape that we can feed to the ConformerBlock. The block, currently, can support an input shape of [Batch_Size, Time, Frequency] (in the case of audio). Assume that my Time dimension is not fixed, but is varying. Therefore, an example could be that i get [None(batch_size), None(Time_size), 128(Frequency)]. The block as it is does not support it. The error is occurring in the attention.py that the line seq = tf.range(n) does not support None values. Any ideas how to overcome this issue?

anasvaf avatar Apr 19 '23 11:04 anasvaf