From 80feda749df56cb8efbef50a0d6459da8f4cf776 Mon Sep 17 00:00:00 2001
From: VectorSL <shiliang10@huawei.com>
Date: Fri, 18 Dec 2020 14:28:45 +0800
Subject: [PATCH] fix init faild

---
 model_zoo/official/cv/resnet/src/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/model_zoo/official/cv/resnet/src/dataset.py b/model_zoo/official/cv/resnet/src/dataset.py
index b0a2ebad97..df19ed472c 100755
--- 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: