Is there any necessity to have the total number of files to be multiple of batch_size in pytorch? I am having multiple folders containing NumPy files, but each folder is not a multiple of 32 and I am getting size mismatch error while training the network? Is there any fix or tips on loading data so that size mismatch does not occur?