Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Commit 207a08f

Browse files
authored
Jfinks sparsezoo doc update (#124) (#127)
1 parent c51a70c commit 207a08f

File tree

2 files changed

+6
-39
lines changed

2 files changed

+6
-39
lines changed

README.md

Lines changed: 4 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ limitations under the License.
5353

5454
## Overview
5555

56-
SparseZoo is a constantly-growing repository of sparsified (pruned and pruned-quantized) models with matching sparsification recipes for neural networks.
56+
[SparseZoo is a constantly-growing repository](https://sparsezoo.neuralmagic.com) of sparsified (pruned and pruned-quantized) models with matching sparsification recipes for neural networks.
5757
It simplifies and accelerates your time-to-value in building performant deep learning models with a collection of inference-optimized models and recipes to prototype from.
5858
Read more about sparsification [here](https://docs.neuralmagic.com/main/source/getstarted.html#sparsification).
5959

@@ -66,8 +66,9 @@ The [GitHub repository](https://github.com/neuralmagic/sparsezoo) contains the P
6666

6767
## Highlights
6868

69-
- [Available Models Listing](https://github.com/neuralmagic/sparsezoo/blob/main/docs/source/models.md)
70-
- [Available Recipes Listing](https://github.com/neuralmagic/sparsezoo/blob/main/docs/source/recipes.md)
69+
- [Model Stub Architecture Overview](https://github.com/neuralmagic/sparsezoo/blob/main/docs/source/models.md)
70+
- [Available Model Recipes](https://github.com/neuralmagic/sparsezoo/blob/main/docs/source/recipes.md)
71+
- [sparsezoo.neuralmagic.com](https://sparsezoo.neuralmagic.com)
7172

7273
## Installation
7374

@@ -82,41 +83,6 @@ pip install sparsezoo
8283

8384
## Quick Tour
8485

85-
Each model in the SparseZoo has a specific stub that identifies it. The stubs are made up of the following structure:
86-
87-
`DOMAIN/SUB_DOMAIN/ARCHITECTURE{-SUB_ARCHITECTURE}/FRAMEWORK/REPO/DATASET{-TRAINING_SCHEME}/SPARSE_NAME-SPARSE_CATEGORY-{SPARSE_TARGET}`
88-
89-
The properties within each model stub are defined as the following:
90-
91-
| Model Property | Definition | Examples |
92-
|:----------------:|:----------------------------------------------------------------------------------------------:|:----------------------------------------------------------------------------------:|
93-
| DOMAIN | The type of solution the model is architected and trained for | cv, nlp |
94-
| SUB_DOMAIN | The sub type of solution the model is architected and trained for | classification, segmentation |
95-
| ARCHITECTURE | The name of the guiding setup for the network's graph | resnet_v1, mobilenet_v1 |
96-
| SUB_ARCHITECTURE | (optional) The scaled version of the architecture such as width or depth | 50, 101, 152 |
97-
| FRAMEWORK | The machine learning framework the model was defined and trained in | pytorch, tensorflow_v1 |
98-
| REPO | The model repository the model and baseline weights originated from | sparseml, torchvision |
99-
| DATASET | The dataset the model was trained on | imagenet, cifar10 |
100-
| TRAINING_SCHEME | (optional) A description on how the model was trained | augmented, lower_lr |
101-
| SPARSE_NAME | An overview of what was done to sparsify the model | base, pruned, quant (quantized), pruned_quant, arch (architecture modified) |
102-
| SPARSE_CATEGORY | Descriptor on the degree to which the model is sparsified as compared with the baseline metric | none, conservative (100% baseline), moderate (>= 99% baseline), aggressive (< 99%) |
103-
| SPARSE_TARGET | (optional) Descriptor for the target environment the model was sparsified for | disk, edge, deepsparse, gpu |
104-
105-
The contents of each model are made up of the following:
106-
107-
- model.md: The model card containing metadata, descriptions, and information for the model.
108-
- model.onnx: The [ONNX](https://onnx.ai/) representation of the model's graph.
109-
- model.onnx.tar.gz: A compressed format for the ONNX file.
110-
Currently ONNX does not support sparse tensors and quantized sparse tensors well for compression.
111-
- [FRAMEWORK]/model.[EXTENSION]: The native ML framework file(s) for the model in which it was originally trained.
112-
Such as PyTorch, Keras, TensorFlow V1
113-
- recipes/original.[md|yaml]: The original sparsification recipe used to create the model.
114-
- recipes/[NAME].[md|yaml]: Additional sparsification recipes that can be used with the model such as transfer learning.
115-
- sample-originals: The original sample data without any preprocessing for use with the model.
116-
- sample-inputs: The sample data after pre processing for use with the model.
117-
- sample-outputs: The outputs after running the sample inputs through the model.
118-
- sample-labels: The labels that classify the sample inputs.
119-
12086
### Python APIS
12187

12288
The Python APIs respect this format enabling you to search and download models. Some code examples are given below.

docs/index.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818
.. toctree::
1919
:maxdepth: 3
2020
:caption: General
21-
21+
22+
SparseZoo Repository <https://github.com/neuralmagic/sparsezoo/issues>
2223
source/models
2324
source/recipes
2425

0 commit comments

Comments
 (0)