Browse Source

!13759 fix ST failure of resnet thor of r1.2

From: @wangmin0104
Reviewed-by: @kisnwang,@wang_zi_dong
Signed-off-by: @wang_zi_dong
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
12db983888
3 changed files with 2 additions and 2 deletions
  1. +0
    -0
      mindspore/lite/micro/example/mnist_stm32f746/mnist/benchmark/load_input.c
  2. +0
    -0
      mindspore/lite/micro/example/mnist_stm32f746/mnist/benchmark/load_input.h
  3. +2
    -2
      tests/st/networks/models/resnet50/src_thor/dataset.py

+ 0
- 0
mindspore/lite/micro/example/mnist_stm32f746/mnist/benchmark/load_input.c View File


+ 0
- 0
mindspore/lite/micro/example/mnist_stm32f746/mnist/benchmark/load_input.h View File


+ 2
- 2
tests/st/networks/models/resnet50/src_thor/dataset.py View File

@@ -72,8 +72,8 @@ def create_dataset(dataset_path, do_train, repeat_num=1, batch_size=32):

type_cast_op = C2.TypeCast(mstype.int32)

data_set = data_set.map(operations=trans, input_columns="image", num_parallel_workers=8)
data_set = data_set.map(operations=type_cast_op, input_columns="label", num_parallel_workers=8)
data_set = data_set.map(operations=trans, input_columns="image", num_parallel_workers=16)
data_set = data_set.map(operations=type_cast_op, input_columns="label", num_parallel_workers=16)

# apply batch operations
data_set = data_set.batch(batch_size, drop_remainder=True)


Loading…
Cancel
Save