Browse Source

!12477 [MSLITE] modify train convert name

From: @zhengjun10
Reviewed-by: @HilbertDavid
Signed-off-by: @HilbertDavid
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
3b74db3fad
4 changed files with 9 additions and 9 deletions
  1. +2
    -2
      mindspore/lite/CMakeLists.txt
  2. +2
    -2
      mindspore/lite/examples/train_lenet/model/prepare_model.sh
  3. +2
    -2
      mindspore/lite/examples/transfer_learning/model/prepare_model.sh
  4. +3
    -3
      mindspore/lite/test/run_net_train.sh

+ 2
- 2
mindspore/lite/CMakeLists.txt View File

@@ -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()


+ 2
- 2
mindspore/lite/examples/train_lenet/model/prepare_model.sh View File

@@ -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


+ 2
- 2
mindspore/lite/examples/transfer_learning/model/prepare_model.sh View File

@@ -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


+ 3
- 3
mindspore/lite/test/run_net_train.sh View File

@@ -38,13 +38,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}


Loading…
Cancel
Save