Skip to content

Commit b1535b4

Browse files
authored
docs: add MNIST magnitude pruning tutorial (#33)
* docs: add MNIST magnitude pruning tutorial Adds a tutorial notebook demonstrating post-training and fine-tuning-based magnitude pruning on an MNIST model, mirroring the existing quantization and palettization tutorials. Also adds the missing Pruning entry to the sidebar toctree. * docs: use 70% sparsity for post-training pruning in MNIST tutorial Aligns the post-training pruning section with the fine-tuning section, which already targets 70% sparsity, instead of the default 50%. * docs: use 50% sparsity in MNIST pruning tutorial, bake in outputs Aligns post-training and fine-tuning sparsity at 50% and ships the notebook with executed cell outputs, showing the actual effect of pruning on accuracy end-to-end. * docs: strip coremltools import warning from pruning tutorial output Removes a local scikit-learn/coremltools version warning baked into the notebook's output; it's environment noise, not tutorial content. * docs: fix broken pruning config link in MNIST pruning tutorial
1 parent d1e5d37 commit b1535b4

3 files changed

Lines changed: 739 additions & 0 deletions

File tree

docs/src/examples/toy_models.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,5 @@ These tutorials demonstrate model compression techniques using simple toy models
88
/tutorials/mnist_quantization
99
/tutorials/mnist_palettization
1010
/tutorials/mnist_palettization_and_activation_quantization
11+
/tutorials/mnist_pruning
1112
```

docs/src/index.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ examples/model_examples
2929
3030
quantization/index
3131
palettization/index
32+
pruning/index
3233
```
3334

3435
```{toctree}

0 commit comments

Comments
 (0)