Update Example for Pytorch 3x Mixed Precision
Type of Change
Update example for Pytorch 3x mixed precision
Description
- [x] add Torchvision resnet18 model as an example
- [x] update document
How has this PR been tested?
extension test
Dependency Change?
No
⚡ Required checks status: All passing 🟢
Groups summary
🟢 Code Scan Tests workflow
| Check ID | Status | Error details | |
|---|---|---|---|
| Code-Scan | success | ✅ | |
| Code-Scan (Bandit Code Scan Bandit) | success | ✅ | |
| Code-Scan (DocStyle Code Scan DocStyle) | success | ✅ | |
| Code-Scan (Pylint Code Scan Pylint) | success | ✅ |
These checks are required after the changes to neural_compressor/common/utils/constants.py, neural_compressor/torch/algorithms/mixed_precision/__init__.py, neural_compressor/torch/algorithms/mixed_precision/half_precision_convert.py, neural_compressor/torch/algorithms/mixed_precision/module_wrappers.py, neural_compressor/torch/quantization/__init__.py, neural_compressor/torch/quantization/algorithm_entry.py, neural_compressor/torch/quantization/config.py.
🟢 Model Tests 3x workflow
| Check ID | Status | Error details | |
|---|---|---|---|
| Model-Test-3x | success | ✅ | |
| Model-Test-3x (Generate Report GenerateReport) | success | ✅ | |
| Model-Test-3x (Run PyTorch Model opt_125m_woq_gptq_int4) | success | ✅ | |
| Model-Test-3x (Run PyTorch Model opt_125m_woq_gptq_int4_dq_bnb) | success | ✅ | |
| Model-Test-3x (Run PyTorch Model opt_125m_woq_gptq_int4_dq_ggml) | success | ✅ |
These checks are required after the changes to neural_compressor/common/utils/constants.py, neural_compressor/torch/algorithms/mixed_precision/__init__.py, neural_compressor/torch/algorithms/mixed_precision/half_precision_convert.py, neural_compressor/torch/algorithms/mixed_precision/module_wrappers.py, neural_compressor/torch/quantization/__init__.py, neural_compressor/torch/quantization/algorithm_entry.py, neural_compressor/torch/quantization/config.py.
🟢 Unit Tests 3x-TensorFlow workflow
| Check ID | Status | Error details | |
|---|---|---|---|
| UT-3x-TensorFlow | success | ✅ | |
| UT-3x-TensorFlow (Coverage Compare CollectDatafiles) | success | ✅ | |
| UT-3x-TensorFlow (Unit Test 3x TensorFlow Unit Test 3x TensorFlow) | success | ✅ | |
| UT-3x-TensorFlow (Unit Test 3x TensorFlow baseline Unit Test 3x TensorFlow baseline) | success | ✅ |
These checks are required after the changes to neural_compressor/common/utils/constants.py.
🟢 Unit Tests 3x-PyTorch workflow
| Check ID | Status | Error details | |
|---|---|---|---|
| UT-3x-Torch | success | ✅ | |
| UT-3x-Torch (Coverage Compare CollectDatafiles) | success | ✅ | |
| UT-3x-Torch (Unit Test 3x Torch Unit Test 3x Torch) | success | ✅ | |
| UT-3x-Torch (Unit Test 3x Torch baseline Unit Test 3x Torch baseline) | success | ✅ |
These checks are required after the changes to neural_compressor/common/utils/constants.py, neural_compressor/torch/algorithms/mixed_precision/__init__.py, neural_compressor/torch/algorithms/mixed_precision/half_precision_convert.py, neural_compressor/torch/algorithms/mixed_precision/module_wrappers.py, neural_compressor/torch/quantization/__init__.py, neural_compressor/torch/quantization/algorithm_entry.py, neural_compressor/torch/quantization/config.py, test/3x/torch/test_autotune.py.
Thank you for your contribution! 💜
Note This comment is automatically generated and will be updates every 180 seconds within the next 6 hours. If you have any other questions, contact chensuyue or XuehaoSun for help.
Need to rebase master so incbench can be available in this PR.