Hey guys,
I am facing a quite strange error, which I have not encountered before.
I have a data loader with batches of size 256. My training begins and everything is fine, meaning that I have the correct sizes for batch_x and and batch_y, being [256, 1, 28, 28] and [256] respectively.
Then something happens and these sizes become [96, 1, 28, 28] and [96].
Any ideas how this is even possible?