!24168 [MS][LITE]fix train code example

Merge pull request !24168 from yefeng/166-code_example
This commit is contained in:
i-robot 2021-09-27 11:59:02 +00:00 committed by Gitee
commit d22f621eea
1 changed files with 1 additions and 1 deletions

View File

@ -407,7 +407,7 @@ function Run_CodeExamples() {
export PATH=${x86_path}/mindspore-lite-${version}-linux-x64/tools/converter/converter/:$PATH
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${x86_path}/mindspore-lite-${version}-linux-x64/tools/converter/lib/:${x86_path}/mindspore-lite-${version}-linux-x64/tools/converter/third_party/glog/lib
if [[ "$should_run" == "1" && ($backend == "all" || $backend == "x86-all" || $backend == "x86_train" || $backend == "x86-java") ]]; then
if [[ "$should_run" == "1" && ($backend == "all" || $backend == "codegen_and_train" || $backend == "x86-all" || $backend == "x86_train" || $backend == "x86-java") ]]; then
cd ${basepath}/../../examples/train_lenet_java || exit 1
chmod 777 ./prepare_and_run.sh
./prepare_and_run.sh -D ${datasets_path}/mnist -r ${tarball_path} -m ${models_path}/code_example.mindir >> ${run_code_examples_log_file}