Simplifier failure: [ONNXRuntimeError]

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.33k stars 4.2k forks source link

Simplifier failure: [ONNXRuntimeError] #132

Open banhongjun opened 2 years ago

banhongjun commented 2 years ago

Simplifier failure: [ONNXRuntimeError] : 1 : FAIL : Node (Mul_658) Op (Mul) [ShapeInferenceError] Incompatible dimensions.

philipp-schmidt commented 2 years ago

check pytorch version and use the most recent one

sarmientoj24 commented 2 years ago

python models/export.py --weights yolov7.pt --grid
Namespace(batch_size=1, device='cpu', dynamic=False, grid=True, img_size=[640, 640], weights='yolov7.pt')
YOLOR 🚀 v0.1-28-g6bc5a4d torch 1.10.0+cu102 CPU

But i still get that error