Hi Im facing the same problem even though I put droplast=True on a dataloader and also multilple batch size by number of GPU.
