qq_40934825的博客

在深度学习的训练过程中，我们需要将数据分批的放入到训练网络中，批数量的大小也被成为batch_size,通过pytorch提供的dataloader方法，可以自动实现一个迭代器，每次返回一组batch_size个样本和标签来进行训练。下面是一个dataloader的简单例子：import torchimport torch.utils.data as DataBATCH_SIZE = 5#生成1,2,3,4,5,6,7,8,9,10x = torch.linspace(1, 10, 10)#