Browse Source

Fix dtype problem

tags/v1.1.0
xiefangqi 5 years ago
parent
commit
e9c2bc7099
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

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


Loading…
Cancel
Save