Has anyone tried changing the batch size

[sammens]

I am having issues trying to change the batch size. The code produces a mismatch error each time I change the batch size. Below is the error I am getting;
RuntimeError: size mismatch, m1: [4 x 8192], m2: [2048 x 5] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:283

[iyaja]

I was able to run the CIFAR10 experiments with different batch sizes using the --batch-size flag without any issues.