This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
Fix dtype problem
tags/v1.1.0
xiefangqi
5 years ago
parent
e6960ba7fd
commit
e9c2bc7099
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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 == "f
loat
16":
if dtype == "f
p
16":
normalize_op = C.NormalizePad(mean=mean, std=std, dtype="float16")
if do_train:
trans = [
Write
Preview
Loading…
Cancel
Save