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

Commit 08abc66

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

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

alficore/dataloader/abs_loader.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,13 @@
99
import os
1010
import random
1111
import torch
12+
import sys, re
1213
from alficore.ptfiwrap_utils.helper_functions import TEM_Dataloader_attr
1314
from .objdet_baseClasses.catalog import DatasetCatalog, MetadataCatalog
1415
from .objdet_baseClasses.common import DatasetFromList, MapDataset, DatasetMapper, trivial_batch_collator
1516

17+
python_version_str = sys.version
18+
1619
class Abstract_Loader(ABC):
1720
def __init__(self, dl_attr:TEM_Dataloader_attr, dnn_model_name:str="yolov3_ultra"):
1821
"""
@@ -144,8 +147,6 @@ def register_instances(self, json_file, image_root, dataset_name, metadata={}, d
144147

145148
def datagen_itr(self):
146149
if self.data_incoming == True:
147-
import sys, re
148-
python_version_str = sys.version
149150
# Use regular expressions to extract the version number
150151
version_number = re.search(r'\d+\.\d+\.\d+', python_version_str).group()
151152
version = version_number.group()

0 commit comments

Comments
 (0)