Update stf/stf-api-alternative/src/stf_alternative/readers.py

stf/stf-api-alternative/src/stf_alternative/readers.py

@@ -44,7 +44,9 @@ class ImageDataset(Dataset):
 
 class ProcessPoolIterator:
     def __init__(self, dataset, preload=8, num_workers=2):
-        self.pool = ProcessPoolExecutor(num_workers)
+        #self.pool = ProcessPoolExecutor(num_workers)
+        self.pool = ThreadPoolExecutor(num_workers)
+        
         self.dataset = dataset
         self.queue = deque()
         self.preload = preload
@@ -95,7 +97,8 @@ class ProcessPoolBatchIterator:
 
 class AsyncProcessPoolIterator:
     def __init__(self, dataset, preload=8, num_workers=4):
-        self.pool = ProcessPoolExecutor(num_workers)
+        #self.pool = ProcessPoolExecutor(num_workers)
+        self.pool = ThreadPoolExecutor(num_workers)
         self.dataset = dataset
         self.queue = deque()
         self.preload = preload