the pad_batch function is error?

[quoniammm]

I feel that np.max should be used.However, here is np.mean

[Sandeep42]

Yeah, np.max should be used ideally. I used mean because it is consuming too much of GPU memory. Now, I think you should ideally sort samples with their lengths so that the padding could be minimised. That hasn't been implemented here. You can have a look here.

https://github.com/EdGENetworks/anuvada