add train ut

This commit is contained in:
lz 2020-10-15 16:57:01 +08:00
parent 1e678a84dc
commit 4e770695cb
2 changed files with 11 additions and 0 deletions

View File

@ -0,0 +1,5 @@
#!/bin/bash
cd ./ut/src/runtime/kernel/arm || exit 1
../../../../../../build/test/lite-test --gtest_filter=NetworkTest.efficient_net
../../../../../../build/test/lite-test --gtest_filter=NetworkTest.tuning_layer
../../../../../../build/test/lite-test --gtest_filter=NetworkTest.lenetnet

6
mindspore/lite/test/runtest.sh Executable file
View File

@ -0,0 +1,6 @@
#!/bin/bash
cd ./mindspore/lite/test/ || exit 1
if [ $1 == train ]; then
echo 'run train ut tests'
./run_train_ut.sh
fi