已开启
fix _MultiProcessingDataLoaderIter object has no attribute _in_order #195
kuhn7创建于 2月14日
fix _MultiProcessingDataLoaderIter object has no attribute _in_order #195
已开启
共 1 个文件变更+3-0
| @@ -95,7 +95,10 @@ class _MultiProcessingDataLoaderIter(SrcMultiProcessingDataLoaderIter): | |||
| 95 | torch_npu.npu.synchronize() | 95 | torch_npu.npu.synchronize() |
| 96 | except Exception as e: | 96 | except Exception as e: |
| 97 | print(e) | 97 | print(e) |
| 98 | + | ||
| 98 | self._prefetch_factor = loader.prefetch_factor | 99 | self._prefetch_factor = loader.prefetch_factor |
| 100 | + self._in_order = loader.in_order | ||
| 101 | + | ||
| 99 | self._dataset = loader.dataset | 102 | self._dataset = loader.dataset |
| 100 | self._shared_seed = None | 103 | self._shared_seed = None |
| 101 | self._pg = None | 104 | self._pg = None |