Hi there,
I load the model from torchvision.models.vgg16_bn(pretrained=True).
When forward the image batch to extract the feature, it leads to RunTimeError: size mismatch at /opt/conda/conda-bld/pytorch_1503966894950/work/torch/lib/THC/generic/THCTensorMathBlas.cu:243
The shape of input is torch.Size([50, 3, 224, 224]).
So… what’s wrong with this?