train_imagenet_cpu_base.sh 292 字节
Newer Older
1 2 3 4 5 6 7 8 9
CUDA_VISIBLE_DEVICES='' python train.py \
          --batch_size 64 \
          --data imagenet \
          --pruning_mode ratio \
          --ratio 0.55 \
          --lr 0.05 \
          --model MobileNet \
          --pretrained_model "MobileNetV1_pretrained" \
          --use_gpu False \