-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Documentation for the expected input dimension of the model class #8777
Description
📚 The doc issue
The built-in models are really convenient. However, the documentation usually did not specified the expected input dimension, I always find it troublesome to confirm what is the correct input dimension for the model class that i want to use.
For example:
https://pytorch.org/vision/main/models/generated/torchvision.models.resnet18.html
https://pytorch.org/vision/main/models/generated/torchvision.models.swin_t.html
https://pytorch.org/vision/main/models/generated/torchvision.models.video.swin3d_b.html
Is there clear documentation for this issue? Or is there a simple and clear rule that i can use (e.g., a rule that were used to develop these model class in pytorch that are consistent throughout?)
Suggest a potential alternative/fix
No response