Update protenix/data/infer_data_pipeline.py
Browse files
protenix/data/infer_data_pipeline.py
CHANGED
|
@@ -61,7 +61,7 @@ def get_inference_dataloader(configs: Any) -> DataLoader:
|
|
| 61 |
batch_size=1,
|
| 62 |
sampler=sampler,
|
| 63 |
collate_fn=lambda batch: batch,
|
| 64 |
-
num_workers=
|
| 65 |
)
|
| 66 |
return dataloader
|
| 67 |
|
|
|
|
| 61 |
batch_size=1,
|
| 62 |
sampler=sampler,
|
| 63 |
collate_fn=lambda batch: batch,
|
| 64 |
+
num_workers=0,
|
| 65 |
)
|
| 66 |
return dataloader
|
| 67 |
|