RuntimeError: Given groups=1, weight of size [32, 3, 3, 3], expected input[1, 224, 4, 225] to have 3 channels, but got 224 channels instead #45
Description
- Classifier Model trained on tensorflow efficientnetb0 architecture on 224*224 images with 3 channels and this model converted to .onnx format.
- Then, .onnx to pytorch model in .pth format.
- While loading a model on 224*224 images getting an error like
RuntimeError: Given groups=1, weight of size [32, 3, 3, 3], expected input[1, 224, 4, 225] to have 3 channels, but got 224 channels instead