Skip to content

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
Contributor

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