A automatic pipeline for SnapML quantization
Project description
# SnapML Automatic Quantization Pipeline
## Installation:
pip install git+ssh://git@github.sc-corp.net/Snapchat/snapml_auto_quantization.git –upgrade
## Uninstallation:
pip uninstall snapml_auto_quantization
## Exporting ONNX Models:
- torch.onnx.export(net, inputs, onnx_path, verbose=verbose,
input_names=input_names, output_names=output_names, keep_initializers_as_inputs=True, operator_export_type= torch.onnx.OperatorExportTypes.ONNX_ATEN_FALLBACK, opset_version=None)
## API Documenation
[Documentation](https://github.sc-corp.net/pages/Snapchat/snapml_auto_quantization/)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file snapml_auto_quantization-0.0.2.tar.gz
.
File metadata
- Download URL: snapml_auto_quantization-0.0.2.tar.gz
- Upload date:
- Size: 17.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.1
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3df3472087fe2f4a3c18107182b232e1ed226511fa24263ecd1c39f08d4e1429 |
|
MD5 | 9329ff37a1775e3aec13490e2d53d3d0 |
|
BLAKE2b-256 | d4e85890d9c2c27546e1910713ac7db59e5b088c2bfa40e5eed48189d01b0609 |