fix optimization level in vision.py script example (#268)

neuralmagic · Jun 2, 2021 · f2e2a37 · f2e2a37
1 parent d72c73e
commit f2e2a37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/pytorch/vision.py b/integrations/pytorch/vision.py
@@ -368,7 +368,7 @@
 python integrations/pytorch/vision.py train \
     --sparse-transfer-learn \
     --recipe-path  ~/sparseml_recipes/pruning_mobilenet.yaml \
-    --arch-key mobilenet_v1 --pretrained recal-perf \
+    --arch-key mobilenet_v1 --pretrained pruned-moderate \
     --dataset imagefolder --dataset-path ~/datasets/my_imagefolder_dataset \
     --train-batch-size 256 --test-batch-size 1024