PyTorch DataLoader 和并行性

Question

我创建了一个扩展 DataSet 的类来为分割任务加载图像，因此一个输入和一个输出。每次调用 getitem 方法时，该类都会对输入和输出进行必要的数据扩充操作，并且运行良好。

但是，当我将此类与 PyTorch DataLoader 一起使用时，输入转换与输出转换不​​匹配。要执行相同的操作，我必须获取/设置随机操作/类的状态，我敢打赌 DataLoader 会这样做，因此它们之间存在冲突。

我该如何解决？

Answer 1

解决方案是创建所有 Random 类使用的本地实例，因为 DataLoader 没有。这样做，所有执行的随机转换都是根据不受 DataLoader 影响的随机值/状态。常见的做法似乎是创建一个类并将所有转换放入其中。