Skip to content

Commit f2e2a37

Browse files
authored
fix optimization level in vision.py script example (#268)
1 parent d72c73e commit f2e2a37

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

integrations/pytorch/vision.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -368,7 +368,7 @@
368368
python integrations/pytorch/vision.py train \
369369
--sparse-transfer-learn \
370370
--recipe-path ~/sparseml_recipes/pruning_mobilenet.yaml \
371-
--arch-key mobilenet_v1 --pretrained recal-perf \
371+
--arch-key mobilenet_v1 --pretrained pruned-moderate \
372372
--dataset imagefolder --dataset-path ~/datasets/my_imagefolder_dataset \
373373
--train-batch-size 256 --test-batch-size 1024
374374

0 commit comments

Comments
 (0)