Skip to content

Commit 6ce9912

Browse files
authored
Merge pull request #284 from Project-MONAI/slbryson/modify-dataloader
Issue (#283) Fix shm issue
2 parents 4ab27de + 4f19ee0 commit 6ce9912

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/deploy/operators/monai_seg_inference_operator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def compute(self, op_input: InputContext, op_output: OutputContext, context: Exe
208208
model = torch.jit.load(MonaiSegInferenceOperator.MODEL_LOCAL_PATH, map_location=device)
209209

210210
dataset = Dataset(data=[{self._input_dataset_key: img_name}], transform=pre_transforms)
211-
dataloader = DataLoader(dataset, batch_size=1, shuffle=False, num_workers=1)
211+
dataloader = DataLoader(dataset, batch_size=1, shuffle=False, num_workers=0)
212212

213213
with torch.no_grad():
214214
for d in dataloader:

0 commit comments

Comments
 (0)