Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

[Model Compression] fix compression doc and update a notebook example #3955

Merged
merged 5 commits into from
Jul 28, 2021

Conversation

J-shang
Copy link
Contributor

@J-shang J-shang commented Jul 19, 2021

No description provided.

@J-shang J-shang requested review from QuanluZhang and linbinskn July 19, 2021 07:54
@QuanluZhang QuanluZhang requested a review from ultmaster July 21, 2021 03:08
@@ -31,7 +31,7 @@ The specification of configuration can be found `here <./Tutorial.rst#specify-th
Step2. Choose a pruner and compress the model
Copy link
Contributor

Choose a reason for hiding this comment

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

I suggest that we can put the notebook into documentation to increase visibility.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

good suggestion, add this into the doc.

@J-shang J-shang closed this Jul 26, 2021
@J-shang J-shang reopened this Jul 26, 2021
}
],
"source": [
"from nni.algorithms.compression.pytorch.quantization import DoReFaQuantizer\n",
Copy link
Contributor

Choose a reason for hiding this comment

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

Recommend using QAT_Quantizer instead of DoReFaQuantizer since quantization speedup only support QAT_Quantizer right now.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

got it, change to QAT_Quantizer

@J-shang J-shang closed this Jul 27, 2021
@J-shang J-shang reopened this Jul 27, 2021
@J-shang J-shang closed this Jul 27, 2021
@J-shang J-shang reopened this Jul 27, 2021
@QuanluZhang QuanluZhang merged commit 3f1e4f5 into microsoft:master Jul 28, 2021
@J-shang J-shang deleted the fix-compression-doc branch August 5, 2021 12:09
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants