From b2a948a5024114a1ed0d1175755eb2c687e3d40a Mon Sep 17 00:00:00 2001 From: zhengjun10 Date: Thu, 11 Mar 2021 16:41:27 +0800 Subject: [PATCH] modify train converter name --- mindspore/lite/CMakeLists.txt | 4 ++-- mindspore/lite/examples/train_lenet/model/prepare_model.sh | 4 ++-- .../lite/examples/transfer_learning/model/prepare_model.sh | 4 ++-- mindspore/lite/test/run_net_train.sh | 6 +++--- 4 files changed, 9 insertions(+), 9 deletions(-) diff --git a/mindspore/lite/CMakeLists.txt b/mindspore/lite/CMakeLists.txt index 143d3798dc..9e124d3832 100644 --- a/mindspore/lite/CMakeLists.txt +++ b/mindspore/lite/CMakeLists.txt @@ -79,11 +79,11 @@ if(SUPPORT_TRAIN) set(RUNTIME_COMPONENT_NAME train-android-aarch32) elseif(WIN32) set(RUNTIME_COMPONENT_NAME train-win-x64) - set(CONVERTER_COMPONENT_NAME train-converter-win-x64) + set(CONVERTER_COMPONENT_NAME converter-win-x64) set(CODEGEN_NAME train-codegen-win-x64) else() set(RUNTIME_COMPONENT_NAME train-linux-x64) - set(CONVERTER_COMPONENT_NAME train-converter-linux-x64) + set(CONVERTER_COMPONENT_NAME converter-linux-x64) set(CODEGEN_NAME train-codegen-linux-x64) endif() else() diff --git a/mindspore/lite/examples/train_lenet/model/prepare_model.sh b/mindspore/lite/examples/train_lenet/model/prepare_model.sh index 9d7929236d..ee60d16aa5 100755 --- a/mindspore/lite/examples/train_lenet/model/prepare_model.sh +++ b/mindspore/lite/examples/train_lenet/model/prepare_model.sh @@ -14,8 +14,8 @@ CONVERTER="../../../build/tools/converter/converter_lite" if [ ! -f "$CONVERTER" ]; then if ! command -v converter_lite &> /dev/null then - tar -xzf ../../../../../output/mindspore-lite-*-train-converter-linux-x64.tar.gz --strip-components 2 --wildcards --no-anchored converter_lite libmindspore_gvar.so - tar -xzf ../../../../../output/mindspore-lite-*-train-converter-linux-x64.tar.gz --strip-components 4 --wildcards --no-anchored libglog.so.0 + tar -xzf ../../../../../output/mindspore-lite-*-converter-linux-x64.tar.gz --strip-components 2 --wildcards --no-anchored converter_lite libmindspore_gvar.so + tar -xzf ../../../../../output/mindspore-lite-*-converter-linux-x64.tar.gz --strip-components 4 --wildcards --no-anchored libglog.so.0 if [ -f ./converter_lite ]; then CONVERTER=./converter_lite else diff --git a/mindspore/lite/examples/transfer_learning/model/prepare_model.sh b/mindspore/lite/examples/transfer_learning/model/prepare_model.sh index 35dd492968..2484c12c53 100755 --- a/mindspore/lite/examples/transfer_learning/model/prepare_model.sh +++ b/mindspore/lite/examples/transfer_learning/model/prepare_model.sh @@ -18,8 +18,8 @@ CONVERTER="../../../build/tools/converter/converter_lite" if [ ! -f "$CONVERTER" ]; then if ! command -v converter_lite &> /dev/null then - tar -xzf ../../../../../output/mindspore-lite-*-train-converter-linux-x64.tar.gz --strip-components 2 --wildcards --no-anchored converter_lite libmindspore_gvar.so - tar -xzf ../../../../../output/mindspore-lite-*-train-converter-linux-x64.tar.gz --strip-components 4 --wildcards --no-anchored libglog.so.0 + tar -xzf ../../../../../output/mindspore-lite-*-converter-linux-x64.tar.gz --strip-components 2 --wildcards --no-anchored converter_lite libmindspore_gvar.so + tar -xzf ../../../../../output/mindspore-lite-*-converter-linux-x64.tar.gz --strip-components 4 --wildcards --no-anchored libglog.so.0 if [ -f ./converter_lite ]; then CONVERTER=./converter_lite else diff --git a/mindspore/lite/test/run_net_train.sh b/mindspore/lite/test/run_net_train.sh index 641531b604..64133a1137 100755 --- a/mindspore/lite/test/run_net_train.sh +++ b/mindspore/lite/test/run_net_train.sh @@ -37,13 +37,13 @@ function Run_Converter() { cd ${x86_path} || exit 1 tar -zxf mindspore-lite-${version}-train-linux-x64.tar.gz || exit 1 - tar -zxf mindspore-lite-${version}-train-converter-linux-x64.tar.gz || exit 1 - cd ${x86_path}/mindspore-lite-${version}-train-converter-linux-x64 || exit 1 + tar -zxf mindspore-lite-${version}-converter-linux-x64.tar.gz || exit 1 + cd ${x86_path}/mindspore-lite-${version}-converter-linux-x64 || exit 1 cp converter/converter_lite ./ || exit 1 # Convert the models - cd ${x86_path}/mindspore-lite-${version}-train-converter-linux-x64 || exit 1 + cd ${x86_path}/mindspore-lite-${version}-converter-linux-x64 || exit 1 rm -rf ${ms_models_path} mkdir -p ${ms_models_path}