Skip to content

Adding multiweight support to Quantized ResNet #4827

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 6 commits into from
Nov 2, 2021

Conversation

datumbox
Copy link
Contributor

@datumbox datumbox commented Nov 2, 2021

Fixes partially #4674

Verified by:

python -u train_quantization.py --test-only --weights ImageNet1K_FBGEMM_RefV1 --model resnet18 --device cpu

cc @bjuncek

@facebook-github-bot
Copy link

facebook-github-bot commented Nov 2, 2021

💊 CI failures summary and remediations

As of commit 442deae (more details on the Dr. CI page):


  • 2/2 failures introduced in this PR

2 failures not recognized by patterns:

Job Step Action
CircleCI binary_linux_conda_py3.7_cu111 packaging/build_conda.sh 🔁 rerun
CircleCI build_docs Build docs 🔁 rerun

1 job timed out:

  • binary_linux_conda_py3.7_cu111

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.

Copy link
Contributor

@jdsgomes jdsgomes left a comment

Choose a reason for hiding this comment

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

LGTM thanks

@datumbox datumbox force-pushed the prototype/quantresnet_multiweight branch from 82182e4 to eaa13dc Compare November 2, 2021 12:54
@datumbox datumbox merged commit 9a34c0c into pytorch:main Nov 2, 2021
@datumbox datumbox deleted the prototype/quantresnet_multiweight branch November 2, 2021 14:35
@datumbox datumbox linked an issue Nov 2, 2021 that may be closed by this pull request
6 tasks
facebook-github-bot pushed a commit that referenced this pull request Nov 6, 2021
Summary:
* Adding multi-weight support to Quantized ResNet.

* Update references script to support testing quantized models with the new API.

* Handle quantized models correctly in ref script.

* Fixing references for quantization.

Reviewed By: kazhang

Differential Revision: D32216675

fbshipit-source-id: 1af009340a85abe24f714c426d5775365b43bfcc
cyyever pushed a commit to cyyever/vision that referenced this pull request Nov 16, 2021
* Adding multi-weight support to Quantized ResNet.

* Update references script to support testing quantized models with the new API.

* Handle quantized models correctly in ref script.

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

Successfully merging this pull request may close these issues.

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