pytorch - 在 CustomDataset 内部或 CustomDataloader 内部进行转换 - 哪个更好？火炬

Question

我是想在 Pytorch 的CustomDataloader类中应用转换来增加数据，还是应该在CustomDataset类中进行？如果您能举例说明如何正确使用它，我将不胜感激。

标签： pytorchdatasetdataloader

Answer 1

为了获得更大的灵活性，这应该在数据集中应用转换和扩充，而不是数据加载器。后者仅用于对批次元素进行采样和组装。