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 = [