Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

fix init faild

tags/v1.1.0
VectorSL 5 years ago
parent
b741fe46b3
commit
80feda749d
1 changed files with 1 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      model_zoo/official/cv/resnet/src/dataset.py

+ 1
- 0
model_zoo/official/cv/resnet/src/dataset.py View File

@@ -169,6 +169,7 @@ def create_dataset3(dataset_path, do_train, repeat_num=1, batch_size=32, target=
device_num = get_group_size() device_num = get_group_size()
else: else:
device_num = 1 device_num = 1
rank_id = 1
if device_num == 1: if device_num == 1:
ds = de.ImageFolderDataset(dataset_path, num_parallel_workers=8, shuffle=True) ds = de.ImageFolderDataset(dataset_path, num_parallel_workers=8, shuffle=True)
else: else:


Write Preview
Loading…
Cancel
Save