Browse Source

!10011 [Model_zoo][MD] Fix model zoo gpu_bench_mark dtype problem

From: @xiefangqi
Reviewed-by: @pandoublefeng,@heleiwang
Signed-off-by: @heleiwang
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
b0734e81bb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/resnet/gpu_resnet_benchmark.py

+ 1
- 1
model_zoo/official/cv/resnet/gpu_resnet_benchmark.py View File

@@ -72,7 +72,7 @@ def create_dataset(dataset_path, do_train, repeat_num=1, batch_size=32, target="

# define map operations
normalize_op = C.Normalize(mean=mean, std=std)
if dtype == "float16":
if dtype == "fp16":
normalize_op = C.NormalizePad(mean=mean, std=std, dtype="float16")
if do_train:
trans = [


Loading…
Cancel
Save