diff --git a/integrations/pytorch/vision.py b/integrations/pytorch/vision.py
index 0b9fbfae44e..294cfdd49d4 100644
--- 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