Spaces:

Zaixi
/

FoldMark

Sleeping

Zaixi commited on Feb 8

Commit

fe8f637

1 Parent(s): e20fdcf

worker=0

Files changed (1) hide show

runner/inference.py CHANGED Viewed

@@ -296,7 +296,7 @@ def infer_detect(runner: InferenceRunner, configs: Any) -> None:
             test_dataset,
             batch_size=1,
             shuffle=False,
-            num_workers=configs.data.num_dl_workers,
             sampler=test_sampler,
             collate_fn=lambda batch: batch[0],
         )

             test_dataset,
             batch_size=1,
             shuffle=False,
+            num_workers=0,
             sampler=test_sampler,
             collate_fn=lambda batch: batch[0],
         )