diff --git a/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py b/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py index f678476ed9..a284925778 100644 --- a/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py +++ b/model_zoo/official/cv/resnet/gpu_resnet_benchmark.py @@ -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 = [