!20217 [MS][LITE][TOD] Train Lenet Java code example -- allow also mnist relative path

Merge pull request !20217 from ehaleva/r1_3_additions
This commit is contained in:
i-robot 2021-07-14 01:17:02 +00:00 committed by Gitee
commit 0ffdf65db0
2 changed files with 3 additions and 3 deletions

View File

@ -29,7 +29,7 @@ checkopts()
do
case "${opt}" in
D)
MNIST_DATA_PATH=$OPTARG
MNIST_DATA_PATH=$(realpath $OPTARG)
;;
d)
DOCKER=$OPTARG
@ -46,8 +46,8 @@ checkopts()
}
checkopts "$@"
if [ "$MNIST_DATA_PATH" == "" ]; then
echo "MNIST Dataset directory path was not provided"
if [ "$MNIST_DATA_PATH" == "" ] || [ ! -d "$MNIST_DATA_PATH" ]; then
echo "MNIST Dataset directory path was not provided or wrong path " $MNIST_DATA_PATH
display_usage
exit 1
fi