PyTorch学习笔记之数据加载和预处理

PyTorch 通过 torch.utils.data 对一般常用的数据加载进行了封装,可以很容易地实现多线程数据预读和批量加载。并且 torchvision 已经预先实现了常用图像数据集,包括前面使用过的 CIFAR-10, ImageNet, COCO, MNIST, LSUN 等数据集,可通过 torchvision.datasets 方便的调用。   Dataset Dataset 是一个
相关文章
相关标签/搜索