Stand-Alone-Self-Attention icon indicating copy to clipboard operation
Stand-Alone-Self-Attention copied to clipboard

Published 20 hours ago • verified publisher icon leaderj1001

Has anyone tried changing the batch size

Open sammens opened this issue 4 years ago • 1 comments

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

sammens avatar Aug 06 '20 08:08 sammens

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

iyaja avatar Oct 19 '20 09:10 iyaja