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

Commit d756278

Browse files
committed
Fix getting started yolov5 sparse transfer command
1 parent 068cd3b commit d756278

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/content/get-started/transfer-a-sparsified-model/cv-object-detection.mdx

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -30,21 +30,21 @@ After the command completes, the trained model will reamin sparse, achieve an mA
3030

3131
```bash
3232
$ sparseml.yolov5.train \
33-
--project yolov5l \
34-
--name sparsified \
35-
--data VOC.yaml \
33+
--weights zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned_quant-aggressive_95?recipe_type=transfer_learn \
34+
--recipe zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned_quant-aggressive_95?recipe_type=transfer_learn \
3635
--cfg models_v5.0/yolov5l.yaml \
37-
--weights zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned_quant-aggressive_95?recipe_type=transfer \
3836
--hyp data/hyps/hyp.finetune.yaml \
39-
--recipe recipes/yolov5.transfer_learn_pruned_quantized.md
37+
--data VOC.yaml \
38+
--project yolov5l \
39+
--name sparsified
4040
```
4141

4242
The most important arguments are `--data`, `--weights`, and `--recipe`:
4343
- `--data` specifies the dataset onto which the model will be fine-tuned
4444
- `--weights` specifies the base model used to start the transfer learning process (can be a SparseZoo stub or local custom model path)
4545
- `--recipe` specifies the hyperparameters of the fine-tuning process (can be a SparseZoo stub or a local custom recipe)
4646

47-
To utilize your own dataset, set up the appropriate image dataset structure and pass the path as the `--data` argument.
47+
To utilize your own dataset, set up the appropriate image dataset structure and pass the path as the `--data` argument.
4848
An example for VOC is on [GitHub](https://github.com/neuralmagic/sparseml/blob/main/src/sparseml/yolov5/data/VOC.yaml).
4949

5050
The `--cfg` and `--hyp` are configuration files. You can checkout the examples on [GitHub](https://github.com/neuralmagic/sparseml/tree/main/src/sparseml/yolov5).

0 commit comments

Comments
 (0)