未验证 提交 3765db3d 编写于 作者: W whs 提交者: GitHub

Refine pretrain download (#1960)

1. Remove pretrain directory in data
2. Add download scripts into run.sh
上级 9726da54
root_url="http://paddle-imagenet-models-name.bj.bcebos.com"
MobileNetV1="MobileNetV1_pretrained.zip"
ResNet50="ResNet50_pretrained.zip"
wget ${root_url}/${MobileNetV1}
unzip ${MobileNetV1}
wget ${root_url}/${ResNet50}
unzip ${ResNet50}
#!/usr/bin/env bash
# download pretrain model
root_url="http://paddle-imagenet-models-name.bj.bcebos.com"
MobileNetV1="MobileNetV1_pretrained.zip"
ResNet50="ResNet50_pretrained.zip"
pretrain_dir='./pretrain'
if [ ! -d ${pretrain_dir} ]; then
mkdir ${pretrain_dir}
fi
cd ${pretrain_dir}
if [ ! -f ${MobileNetV1} ]; then
wget ${root_url}/${MobileNetV1}
unzip ${MobileNetV1}
fi
if [ ! -f ${ResNet50} ]; then
wget ${root_url}/${ResNet50}
unzip ${ResNet50}
fi
cd -
# for distillation
#--------------------
......@@ -5,7 +30,7 @@ export CUDA_VISIBLE_DEVICES=0
python compress.py \
--model "MobileNet" \
--teacher_model "ResNet50" \
--teacher_pretrained_model ./data/pretrain/ResNet50_pretrained \
--teacher_pretrained_model ./pretrain/ResNet50_pretrained \
--compress_config ./configs/mobilenetv1_resnet50_distillation.yaml
......@@ -14,7 +39,7 @@ python compress.py \
#export CUDA_VISIBLE_DEVICES=0
#python compress.py \
#--model "MobileNet" \
#--pretrained_model ./data/pretrain/MobileNetV1_pretrained \
#--pretrained_model ./pretrain/MobileNetV1_pretrained \
#--compress_config ./configs/filter_pruning_sen.yaml
# for uniform filter pruning
......@@ -22,7 +47,7 @@ python compress.py \
#export CUDA_VISIBLE_DEVICES=0
#python compress.py \
#--model "MobileNet" \
#--pretrained_model ./data/pretrain/MobileNetV1_pretrained \
#--pretrained_model ./pretrain/MobileNetV1_pretrained \
#--compress_config ./configs/filter_pruning_uniform.yaml
# for quantization
......@@ -31,6 +56,6 @@ python compress.py \
#python compress.py \
#--batch_size 64 \
#--model "MobileNet" \
#--pretrained_model ./data/pretrain/MobileNetV1_pretrained \
#--pretrained_model ./pretrain/MobileNetV1_pretrained \
#--compress_config ./configs/quantization.yaml
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册