We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d72c73e commit f2e2a37Copy full SHA for f2e2a37
integrations/pytorch/vision.py
@@ -368,7 +368,7 @@
368
python integrations/pytorch/vision.py train \
369
--sparse-transfer-learn \
370
--recipe-path ~/sparseml_recipes/pruning_mobilenet.yaml \
371
- --arch-key mobilenet_v1 --pretrained recal-perf \
+ --arch-key mobilenet_v1 --pretrained pruned-moderate \
372
--dataset imagefolder --dataset-path ~/datasets/my_imagefolder_dataset \
373
--train-batch-size 256 --test-batch-size 1024
374
0 commit comments