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

Commit 58d9138

Browse files
authored
IC doc fixes (#1577)
1 parent 9752f6a commit 58d9138

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

integrations/torchvision/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ pip install sparseml[torchvision]
4646

4747
Neural Magic has pre-sparsified versions of common Torchvision models such as ResNet-50. These models can be deployed directly or can be fine-tuned onto custom dataset via sparse transfer learning. This makes it easy to create a sparse image classification model trained on your dataset.
4848

49-
[Check out the available models](https://sparsezoo.neuralmagic.com/?domain=cv&sub_domain=classification&page=1)
49+
[Check out the available models](https://sparsezoo.neuralmagic.com/?useCase=classification)
5050

5151
### Recipes
5252

@@ -104,7 +104,7 @@ sparseml.image_classification.train \
104104

105105
For full usage, run:
106106
```bash
107-
sparseml.image_classification --help
107+
sparseml.image_classification.train --help
108108
```
109109

110110
## Quick Start: Sparse Transfer Learning with the CLI
@@ -113,7 +113,7 @@ sparseml.image_classification --help
113113

114114
Sparse Transfer is quite similiar to the typical transfer learning process used to train image classification models, where we fine-tune a checkpoint pretrained on ImageNet onto a smaller downstream dataset. With Sparse Transfer Learning, we simply start the fine-tuning process from a pre-sparsified checkpoint and maintain sparsity while the training process occurs.
115115

116-
In this example, we will fine-tune a 95% pruned version of ResNet-50 ([available in SparseZoo](https://sparsezoo.neuralmagic.com/models/cv%2Fclassification%2Fresnet_v1-50%2Fpytorch%2Fsparseml%2Fimagenet%2Fpruned95_quant-none)) onto ImageNette.
116+
In this example, we will fine-tune a 95% pruned version of ResNet-50 ([available in SparseZoo](https://sparsezoo.neuralmagic.com/models/resnet_v1-50-imagenet-pruned95_quantized?comparison=resnet_v1-50-imagenet-base)) onto ImageNette.
117117

118118
### Kick off Training
119119

0 commit comments

Comments
 (0)