未验证 提交 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 # for distillation
#-------------------- #--------------------
...@@ -5,7 +30,7 @@ export CUDA_VISIBLE_DEVICES=0 ...@@ -5,7 +30,7 @@ export CUDA_VISIBLE_DEVICES=0
python compress.py \ python compress.py \
--model "MobileNet" \ --model "MobileNet" \
--teacher_model "ResNet50" \ --teacher_model "ResNet50" \
--teacher_pretrained_model ./data/pretrain/ResNet50_pretrained \ --teacher_pretrained_model ./pretrain/ResNet50_pretrained \
--compress_config ./configs/mobilenetv1_resnet50_distillation.yaml --compress_config ./configs/mobilenetv1_resnet50_distillation.yaml
...@@ -14,7 +39,7 @@ python compress.py \ ...@@ -14,7 +39,7 @@ python compress.py \
#export CUDA_VISIBLE_DEVICES=0 #export CUDA_VISIBLE_DEVICES=0
#python compress.py \ #python compress.py \
#--model "MobileNet" \ #--model "MobileNet" \
#--pretrained_model ./data/pretrain/MobileNetV1_pretrained \ #--pretrained_model ./pretrain/MobileNetV1_pretrained \
#--compress_config ./configs/filter_pruning_sen.yaml #--compress_config ./configs/filter_pruning_sen.yaml
# for uniform filter pruning # for uniform filter pruning
...@@ -22,7 +47,7 @@ python compress.py \ ...@@ -22,7 +47,7 @@ python compress.py \
#export CUDA_VISIBLE_DEVICES=0 #export CUDA_VISIBLE_DEVICES=0
#python compress.py \ #python compress.py \
#--model "MobileNet" \ #--model "MobileNet" \
#--pretrained_model ./data/pretrain/MobileNetV1_pretrained \ #--pretrained_model ./pretrain/MobileNetV1_pretrained \
#--compress_config ./configs/filter_pruning_uniform.yaml #--compress_config ./configs/filter_pruning_uniform.yaml
# for quantization # for quantization
...@@ -31,6 +56,6 @@ python compress.py \ ...@@ -31,6 +56,6 @@ python compress.py \
#python compress.py \ #python compress.py \
#--batch_size 64 \ #--batch_size 64 \
#--model "MobileNet" \ #--model "MobileNet" \
#--pretrained_model ./data/pretrain/MobileNetV1_pretrained \ #--pretrained_model ./pretrain/MobileNetV1_pretrained \
#--compress_config ./configs/quantization.yaml #--compress_config ./configs/quantization.yaml
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册