build dataset loader
概述 在训练机器学习模型中,一般会应用mini-batch的方法来遍历数据集,此外还需要对加载的数据集进行归一化等附加操作。如果能够构建可迭代的数据集模型,并且添加对数据集的附加操作处理接口,那么数据集的使用将非常方便。例如Pytorch提供的对数据集的非常优雅的处理方式:定义附加操作集合,加载数据集。这样即可非常优雅的遍历整个数据集。 transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)) ]) train_loader =...
2017, Aug 25 — 9 minute read