mobilenets-ssd-pytorch icon indicating copy to clipboard operation
mobilenets-ssd-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon tranleanh

incorrect shape when loading model checkpoints

Open neurosynapse opened this issue 2 years ago • 1 comments

Hello,

I have trained several models vgg16-ssd and mb1-ssd-lite. However when I try to run inference on single image and load the model I always get incorrect shapes and weight names. What could cause that?

Examples of incorrect shape and weight names when loading model checkpoints for vgg16-ssd:

image

Best regards, Roberts

neurosynapse avatar Jan 03 '23 11:01 neurosynapse

The same error here :(

theATM avatar Feb 02 '23 23:02 theATM