pytorch/torch/utils/data/_utils
Joel Schlosser 43ea782af3 Multiprocessing support for NT (#110292)
Fixes #110161

Allows NTs to be used in DataLoaders with `num_workers > 1`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110292
Approved by: https://github.com/cpuhrsch, https://github.com/albanD
2023-10-10 21:58:19 +00:00
..
__init__.py [DataLoader] Make distributed lazily initialized & share seed via PG (#85279) 2022-09-23 18:52:52 +00:00
collate.py Multiprocessing support for NT (#110292) 2023-10-10 21:58:19 +00:00
fetch.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
pin_memory.py [BE] Enable ruff's UP rules and autoformat utils/ (#105424) 2023-07-18 20:17:25 +00:00
serialization.py
signal_handling.py
worker.py Revert "reseed all Generators in Dataloader's _worker_loop() -- via GC (#107131)" 2023-08-23 17:08:07 +00:00