Incorrect dimensions for squeeze operator when loading MobileNetV1

When attempting to load the MobileNetV1 model (from Hugging Face’s aidge) during inference, I encounter the following error message:

[ERROR] - Cannot squeeze dimensions with size greater than 1.

Upon checking the tensor dimensions, I observe an unexpected shape: [1, 1001, 9223372036854775809, 1]

added Bug 🐛 label

changed the description

changed title from Incorrect dimensions for qqueeze operator when loading MobileNetV1 to Incorrect dimensions for squeeze operator when loading MobileNetV1

mentioned in merge request !380 (closed)

@noamzerah Please check !380 (closed) and see if it works!

No, unfortunately, this doesn't resolve the issue. I still get the following dimensions for the input tensor: [1, 1001, 9223372036854775809, 1]

After investigation, it seems that the error occurs earlier and propagates through the forwardDims()

Ok, so you will need to continue investigating to find the source of the problem.

Yes, I'm currently tracing the error back, and it seems to originate from avgPooling. I'll update this issue once I've identified the problem!

created branch 261-incorrect-dimensions-for-squeeze-operator-when-loading-mobilenetv1 to address this issue

mentioned in merge request !382 (merged)

This seems fixed. Closing.

closed