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
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