my batch size is 32 for validation and 16 for training with gradient accumulation of 2.
even wihtout grad accum, it’s not working properly.
I even tried with reduced img resolution and batch size 16 for both.