attention-cnn icon indicating copy to clipboard operation
attention-cnn copied to clipboard

Published 20 hours ago • verified publisher icon epfml

Exception size mismatch, m1: [4 x 12288], m2: [512 x 11] at

Open vench opened this issue 4 years ago • 0 comments

Thanks a lot for your work.

I am using code call

model = resnet.ResNet34(num_classes=len(dataimages.label_str)) // len(dataimages.label_str) == 11

But I get this message

size mismatch, m1: [4 x 12288], m2: [512 x 11] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:283__

vench avatar Oct 06 '20 14:10 vench