Skip to content

Adding a list of model's param size( MB ) and layernum of models in torchvision with a standard input shape like 3,224,224 #4996

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Force1ess opened this issue Nov 26, 2021 · 1 comment

Comments

@Force1ess
Copy link

📚 The doc issue

Torchvision is lacking a table of models' size and layernum

Suggest a potential alternative/fix

Recently, I'm working on choosing some models from torchvision.
And as we all know, a model's size and layers are critical to many aspects.
Especially for beginners who don't familiar with these models, this table can help a lot.
If you think this is necessary, I'm glad to help collect these models' size and layernum and pull a request

@datumbox
Copy link
Contributor

Thanks for the proposal. I agree.

@NicolasHug We should consider adding the model size in the meta-data of the new model API and expose it via the documentation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants