From e9c2bc70999e9b6ba0ec6f3cf266e50142cda0d3 Mon Sep 17 00:00:00 2001 From: xiefangqi Date: Tue, 15 Dec 2020 20:28:12 +0800 Subject: [PATCH] Fix dtype problem --- model_zoo/official/cv/resnet/gpu_resnet_benchmark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py b/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py index 757b9564bb..d714d004d1 100644 --- a/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py +++ b/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py @@ -67,7 +67,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 = [