Error when I use torch_model_2_trt.py to convert yolov3-spp.weights

[swim010]

The error is:

RuntimeError: shape '[42, 256, 1, 1]' is invalid for input of size 10751.

Can you help me?

[cmcamdy]

hi，i find that 42x256x1x1=10752>10751
that means u may consider whether the input shape is right,or whether this project based on eriklindernoren is suitable for your model. if your model is based on ultralytics ,that may not work out.