pytorch/torch/utils
SsnL ed19580dc4 Fix dataloader._shutdown_workers if not all workers are started (#23761)
Summary:
Otherwise you may see errors like
```
Exception ignored in: <function _MultiProcessingDataLoaderIter.__del__ at 0x000001F99F5CB9D8>
Traceback (most recent call last):
  File "C:\Users\Divyansh J\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 883, in __del__
    self._shutdown_workers()
  File "C:\Users\Divyansh J\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 860, in _shutdown_workers
    if self.workers_status[worker_id]:
IndexError: list index out of range
```

e.g. https://discuss.pytorch.org/t/how-to-construct-dataset-with-iterator-for-multi-process-dataloader/49612/5
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23761

Differential Revision: D16644687

Pulled By: soumith

fbshipit-source-id: a60e847431264525079456ff422317af1ac2be4b
2019-08-07 09:06:11 -07:00
..
backcompat Simplify python warning settings and cleanup tests. 2017-06-11 05:37:59 -04:00
bottleneck Turn on F401: Unused import warning. (#18598) 2019-03-30 09:01:17 -07:00
data Fix dataloader._shutdown_workers if not all workers are started (#23761) 2019-08-07 09:06:11 -07:00
ffi remove support for c extensions (#12122) 2018-10-01 13:55:28 -07:00
tensorboard Cleanup API and remove 'experimental' warning (#23000) 2019-07-22 12:10:05 -07:00
__init__.py Back out "Revert D15435461: [pytorch][PR] PyTorch ThroughputBenchmark" (#22185) 2019-06-26 16:05:51 -07:00
__init__.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
_cpp_extension_versioner.py arc lint torch/utils (#13141) 2018-10-25 14:59:03 -07:00
checkpoint.py Deprecate variadic inputs of checkpoint_sequential (#21006) 2019-05-28 21:33:45 -07:00
collect_env.py Use system locale in collect_env.py (#22579) 2019-07-07 20:55:31 -07:00
cpp_extension.py Fix build issue. CUDA may be installed in $CUDA_HOME/lib on macOS. (#23491) 2019-07-29 08:08:43 -07:00
dlpack.py arc lint torch/utils (#13141) 2018-10-25 14:59:03 -07:00
file_baton.py Fix python2 and python 3 compatibility found by lint. (#13140) 2018-10-25 17:20:11 -07:00
hooks.py arc lint torch/utils (#13141) 2018-10-25 14:59:03 -07:00
hooks.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
mkldnn.py Add flag to temporarily disable MKL-DNN conv (#23837) 2019-08-06 11:20:26 -07:00
model_zoo.py add/move a few apis in torch.hub (#18758) 2019-04-10 23:10:39 -07:00
throughput_benchmark.py Fix lint error in format_time() in throughput_benchmark.py and clean it up a bit. (#22424) 2019-07-01 22:15:37 -07:00