the pad_batch function is error?
quoniammm opened this issue · comments
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.