I loaded a pretrained ResNet model using the torch vision Library.How do I check the input dimensions of the model?
Basically all pretrained models expect an input of at least [batch_size, 3, 224, 224]
.
For the inception_v3
model you would need [batch_size, 3, 299, 299]
.
1 Like