ValueError: Expected input batch_size (16) to match target batch_size (64)

Hello Rainiefantasy,

I know this is an old issue but have you also managed to resolve this problem? Maybe you remember. I have the exact same problem…

Thank you!