torch_npu.contrib.Prefetcher

torch_npu.contrib.Prefetcher(loader, stream=None)

npu设备上使用的预取程序。

• 参数解释：
  • loder (torch.utils.data.DataLoader or DataLoader like iterator):Using to generate inputs after preprocessing.
  • stream (torch.npu.Stream)：Default None.Because of the limitation of NPU's memory mechanism,if prefetcher is initialized repeatedly during training,a defined stream should be introduced to prevent memory leakage;if prefetcher is initialized only once during training,a defined stream is not necessary.
• 约束条件：

  无

• 示例：

  无