Checking Input Dimensions of a pretrained Model

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