!10166 GPU fix data init

From: @VectorSL Reviewed-by: @pandoublefeng,@limingqi107 Signed-off-by: @limingqi107
4 years ago · 9fa5fa6162
parent 58cb834733 80feda749d
commit 9fa5fa6162
1 changed files with 1 additions and 0 deletions
--- a/model_zoo/official/cv/resnet/src/dataset.py
+++ b/model_zoo/official/cv/resnet/src/dataset.py
@ -169,6 +169,7 @@ def create_dataset3(dataset_path, do_train, repeat_num=1, batch_size=32, target=
            device_num = get_group_size()
        else:
            device_num = 1
            rank_id = 1
    if device_num == 1:
        ds = de.ImageFolderDataset(dataset_path, num_parallel_workers=8, shuffle=True)
    else: