PyTorch 从入门到放弃 —— 加载数据 审核中
这样就可以把数据集封装起来,实现自动分批,取样,打乱和多处理器协同加载。在这里,我们定义每批大小为65,这样一来,分批遍历dataloader的时候,就能在循环中每次取到64组特征和标记。本教程我们使用FashionMNIST数据集。模块包含了各种视觉数据集, 比如 CIFAR, COCO (. 在本教程中,我们使用TorchVision。Dataset是样本数据集,
原创
2023-06-26 17:25:18 ·
61 阅读 ·
1 评论