Skip to content
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

Clean up purely informational fields from Weight Meta-data #5852

Merged
merged 11 commits into from
Apr 21, 2022

Conversation

datumbox
Copy link
Contributor

@datumbox datumbox commented Apr 21, 2022

Continuation of work from #5848 and #5842 to clean up the meta-data of the Multi-weight support API.

This PR:

With the above changes, the meta-data should contain fields that are intended to be manipulated programmatically hence we can guarantee BC. The only exception is the recipe URL which is there for informational purposes and can be changed to a different value on the future but this is need to to build an effective documentation. All purely informational fields, should be moved on the doc strings.

Copy link
Contributor Author

@datumbox datumbox left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some clarifications:

torchvision/models/vision_transformer.py Show resolved Hide resolved
torchvision/models/vision_transformer.py Outdated Show resolved Hide resolved
torchvision/models/regnet.py Show resolved Hide resolved
Copy link
Member

@NicolasHug NicolasHug left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @datumbox , minor Qs but LGTM anyway

torchvision/models/detection/ssd.py Show resolved Hide resolved
torchvision/models/vision_transformer.py Outdated Show resolved Hide resolved
@datumbox datumbox merged commit f1587a2 into pytorch:main Apr 21, 2022
@datumbox datumbox deleted the models/clean_metadata branch April 22, 2022 18:03
facebook-github-bot pushed a commit that referenced this pull request May 5, 2022
…5852)

Summary:
* Removing `task`, `architecture` and `quantization`

* Fix mypy

* Remove size field

* Remove unused import.

* Fix mypy

* Remove size from schema list.

* update todo

* Simplify with assert

* Adding min_size to all models.

* Update RAFT min size to 128

Reviewed By: jdsgomes, NicolasHug

Differential Revision: D36095646

fbshipit-source-id: 80caefd0dd39a250fa61687d6a8e77f709916ddd
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants