fix train ut

This commit is contained in:
lz 2020-10-16 11:13:06 +08:00
parent 4e770695cb
commit a37453a9f8
1 changed files with 5 additions and 2 deletions

View File

@ -1,6 +1,9 @@
#!/bin/bash
cd ./mindspore/lite/test/ || exit 1
set -e
CURRPATH=$(cd "$(dirname $0)"; pwd)
cd ${CURRPATH}
if [ $1 == train ]; then
echo 'run train ut tests'
./run_train_ut.sh
bash run_train_ut.sh
fi