Skip to content

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
Contributor

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