未验证 提交 5c8d654e 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix slim weight path (#2352)

上级 4e2ed8ff
......@@ -2,8 +2,8 @@
# download pretrain model
root_url="http://paddle-imagenet-models-name.bj.bcebos.com"
MobileNetV1="MobileNetV1_pretrained.zip"
ResNet50="ResNet50_pretrained.zip"
MobileNetV1="MobileNetV1_pretrained.tar"
ResNet50="ResNet50_pretrained.tar"
pretrain_dir='./pretrain'
if [ ! -d ${pretrain_dir} ]; then
......@@ -14,12 +14,12 @@ cd ${pretrain_dir}
if [ ! -f ${MobileNetV1} ]; then
wget ${root_url}/${MobileNetV1}
unzip ${MobileNetV1}
tar xf ${MobileNetV1}
fi
if [ ! -f ${ResNet50} ]; then
wget ${root_url}/${ResNet50}
unzip ${ResNet50}
tar xf ${ResNet50}
fi
cd -
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册