Browse Source

adjust test model

tags/v0.7.0-beta
guohongzilong 5 years ago
parent
commit
ad351a2b32
3 changed files with 12 additions and 2 deletions
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp32/split.cc
  2. +1
    -1
      mindspore/lite/test/models_caffe.cfg
  3. +10
    -0
      mindspore/lite/test/models_tflite.cfg

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp32/split.cc View File

@@ -81,7 +81,7 @@ int SplitCPUKernel::Run() {
auto in_tensor = in_tensors_.front();
input_ptr_ = reinterpret_cast<float *>(in_tensor->Data());
for (int i = 0; i < param->num_split_; i++) {
output_ptr_.push_back(reinterpret_cast<float *>(out_tensors_.at(i)->Data()));
output_ptr_[i] = reinterpret_cast<float *>(out_tensors_.at(i)->Data());
}
ret = LiteBackendParallelLaunch(SplitRun, this, thread_n_num_);
if (ret != RET_OK) {


+ 1
- 1
mindspore/lite/test/models_caffe.cfg View File

@@ -5,7 +5,7 @@ gender_res_large_deploy
glasses
hat
isface
ml_bank_detect_0312
#ml_bank_detect_0312
ml_face_div_parsing
ml_hardware_eyeclose
ml_ocr_detect_20200305


+ 10
- 0
mindspore/lite/test/models_tflite.cfg View File

@@ -35,6 +35,16 @@ mnasnet_1.3_224.tflite
inception_v3.tflite
deeplabv3_257_mv_gpu.tflite
multi_person_mobilenet_v1_075_float.tflite
#hiai_vad.tflite
#ide_label_base.tflite
#ide_label_retrained.tflite
#ml_ei_headpose.tflite
#ml_ei_landmark.tflite
#mnist.tflite
#mobilenet.tflite
#resnet.tflite
#scan_hms_angle1.tflite
#scan_hms_detect.tflite
hiai_latin_ocr.tflite
hiai_latin_ocr_1.tflite
ml_ocr_jk.tflite


Loading…
Cancel
Save