Compression Examples
Distiller comes with sample applications and tutorials covering a range of model types:
Model Type | Sparsity | Post-training quantization | Quantization-aware training | Auto Compression (AMC) | In Directories |
---|---|---|---|---|---|
Image classification | ✅ | ✅ | ✅ | ✅ | classifier_compression, auto_compression/amc |
Word-level language model | ✅ | ✅ | word_language_model | ||
Translation (GNMT) | ✅ | GNMT | |||
Recommendation System (NCF) | ✅ | ncf | |||
Object Detection | ✅ | object_detection_compression |
The directories specified in the table contain the code implementing each of the modalities. The rest of the sub-directories in this directory are each dedicated to a specific compression method, and contain YAML schedules and other files that can be used with the sample applications. Most of these files contain details on the results obtained and how to re-produce them.