Skip to content
This repository was archived by the owner on Oct 15, 2025. It is now read-only.

Commit 14241f5

Browse files
committed
fixed multiprocessor dataloader iterator issue for diff python versions
1 parent 99c3f20 commit 14241f5

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

alficore/dataloader/abs_loader.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,16 @@ def register_instances(self, json_file, image_root, dataset_name, metadata={}, d
144144

145145
def datagen_itr(self):
146146
if self.data_incoming == True:
147-
self.data = self.datagen_iter.next()
147+
import sys, re
148+
python_version_str = sys.version
149+
# Use regular expressions to extract the version number
150+
version_number = re.search(r'\d+\.\d+\.\d+', python_version_str).group()
151+
version = version_number.group()
152+
major, minor, patch = map(int, version.split('.'))
153+
if (major, minor, patch) > (3, 9, 18):
154+
self.data = next(self.datagen_iter)
155+
else:
156+
self.data = self.datagen_iter.next()
148157
self.curr_batch_size = len(self.data)
149158
self.datagen_iter_cnt = self.datagen_iter_cnt + self.curr_batch_size
150159
# self.images = self.images.to(self.device)

0 commit comments

Comments
 (0)