Skip to content

Add multi-type support on get_weight() #4967

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

Merged
merged 4 commits into from
Nov 22, 2021

Conversation

datumbox
Copy link
Contributor

@datumbox datumbox commented Nov 20, 2021

Extend the get_weight() to handle properly model builders with multiple acceptable weight enums. This is useful for quantization.

cc @datumbox @bjuncek

@facebook-github-bot
Copy link

facebook-github-bot commented Nov 20, 2021

💊 CI failures summary and remediations

As of commit 4b1f7c2 (more details on the Dr. CI page):


💚 💚 Looks good so far! There are no failures yet. 💚 💚


This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

@datumbox datumbox force-pushed the prototype/multitype_get_weight branch from 9a112e8 to d03a0ee Compare November 20, 2021 14:59
@datumbox datumbox force-pushed the prototype/multitype_get_weight branch from d03a0ee to 7e5ccbf Compare November 20, 2021 15:02
Copy link
Contributor

@prabhat00155 prabhat00155 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! I have a couple of comments.

@datumbox datumbox merged commit 220e0ff into pytorch:main Nov 22, 2021
@datumbox datumbox deleted the prototype/multitype_get_weight branch November 22, 2021 09:00
@datumbox datumbox linked an issue Nov 22, 2021 that may be closed by this pull request
6 tasks
facebook-github-bot pushed a commit that referenced this pull request Nov 30, 2021
Summary:
* Add multi-type support on get_weight.

* Fix bug on method call.

* Adding logging suffix for QAT.

Reviewed By: NicolasHug

Differential Revision: D32694314

fbshipit-source-id: 4631f32d6138610ec8a0eec97c6c87ad4d4c87e8
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.

Multi-pretrained weights: Add model builders for Quantization
3 participants