提交 6930c737 编写于 作者: G gengdongjie

bugfix for resnet50_imagenet pretrained_ckpt

上级 705c71a2
......@@ -47,7 +47,7 @@ then
exit 1
fi
if [ ! -f "$PATH3" ]
if [ $# == 3 ] && [ ! -f "$PATH3" ]
then
echo "error: PRETRAINED_CKPT_PATH=$PATH3 is not a file"
exit 1
......
......@@ -34,13 +34,13 @@ PATH2=$(get_real_path $2)
if [ ! -d $PATH1 ]
then
echo "error: DATASET_PATH=$1 is not a directory"
echo "error: DATASET_PATH=$PATH1 is not a directory"
exit 1
fi
if [ ! -f $PATH2 ]
then
echo "error: CHECKPOINT_PATH=$2 is not a file"
echo "error: CHECKPOINT_PATH=$PATH2 is not a file"
exit 1
fi
......
......@@ -40,7 +40,7 @@ then
exit 1
fi
if [ ! -f "$PATH2" ]
if [ $# == 2 ] && [ ! -f "$PATH2" ]
then
echo "error: PRETRAINED_CKPT_PATH=$PATH2 is not a file"
exit 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册