import torch.multiprocessing

pytorch · Oct 15, 2024 · 01bfd99 · 01bfd99
1 parent f1d00da
commit 01bfd99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchdata/nodes/pin_memory.py b/torchdata/nodes/pin_memory.py
@@ -5,6 +5,7 @@
 from typing import Iterator, Optional
 
 import torch
+import torch.multiprocessing
 from torch._utils import ExceptionWrapper
 
 from torch.utils.data._utils import MP_STATUS_CHECK_INTERVAL