PyTorch DataLoader和并行

Question

我创建了一个类，该类扩展了DataSet来加载用于分段任务的图像，因此有一个输入和一个输出。每次调用getitem方法时，此类都会在输入和输出上执行必要的操作以进行数据扩充，并且效果很好。

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

我该如何解决？

Answer 1

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