yolov5prune
yolov5prune copied to clipboard
RuntimeError: mat1 and mat2 shapes cannot be multiplied
我在backbone中加入了注意力机制,然后在prune.py中运行报这个错误,RuntimeError: mat1 and mat2 shapes cannot be multiplied (1x62 and 16x1),请问有大佬知道怎么调整吗
有偿,有知道的私聊下我感谢各位大佬!