pytorch/torch/utils/data/_utils
Tongzhou Wang 54ce171f16 Fix persistent_workers + pin_memory (#48543)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/48370 https://github.com/pytorch/pytorch/issues/47445

cc emcastillo who authored the original functionality.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/48543

Reviewed By: bdhirsh

Differential Revision: D25277474

Pulled By: ejguan

fbshipit-source-id: 1967002124fb0fff57caca8982bc7df359a059a2
2021-01-08 07:04:10 -08:00
..
__init__.py
collate.py Remove duplicate assignment in collate.py (#40655) 2020-07-06 12:37:59 -07:00
fetch.py
pin_memory.py
signal_handling.py And type annotations for cpp_extension, utils.data, signal_handling (#42647) 2020-08-06 09:42:07 -07:00
worker.py Fix persistent_workers + pin_memory (#48543) 2021-01-08 07:04:10 -08:00