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

bugfix for resnet50_imagenet pretrained_ckpt

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