RuntimeError: mat1 and mat2 shapes cannot be multiplied (4608x2048 and 4096x1024)
frozen30 opened this issue · 1 comments
frozen30 commented
Hello, I encountered the following problems during the test:
RuntimeError: mat1 and mat2 shapes cannot be multiplied (4608x2048 and 4096x1024)
I checked the parameters of the model and found that 1024 is emb_size, 4096 is img_size, and 4608x2048 refers to the size of the [128, 36, 2048] image.
How can I solve this problem?
fartashf commented
Can you provide more details as how you run the code and the stack trace for this error?