yolov5prune icon indicating copy to clipboard operation
yolov5prune copied to clipboard

RuntimeError: mat1 and mat2 shapes cannot be multiplied

Open DiaJB opened this issue 1 year ago • 1 comments

我在backbone中加入了注意力机制,然后在prune.py中运行报这个错误,RuntimeError: mat1 and mat2 shapes cannot be multiplied (1x62 and 16x1),请问有大佬知道怎么调整吗

DiaJB avatar Jun 01 '23 04:06 DiaJB

有偿,有知道的私聊下我感谢各位大佬!

DiaJB avatar Jun 01 '23 04:06 DiaJB