Open banhongjun opened 2 years ago
check pytorch version and use the most recent one
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
Simplifier failure: [ONNXRuntimeError] : 1 : FAIL : Node (Mul_658) Op (Mul) [ShapeInferenceError] Incompatible dimensions.