train_imagenet.sh 340 字节
Newer Older
M
minghaoBD 已提交
1 2 3 4 5 6 7 8 9 10
#!/bin/bash  
export CUDA_VISIBLE_DEVICES=2,3 
export FLAGS_fraction_of_gpu_memory_to_use=0.98
python3.7 train.py \
          --batch_size 256 \
          --data imagenet \
          --pruning_mode ratio \
          --ratio 0.45 \
          --lr 0.075 \
          --pretrained_model /PaddleSlim/demo/pretrained_model/MobileNetV1_pretrained