酷炫鞋垫

文章
5
资源
0
加入时间
2年10月18天

torch.utils.data.DataLoader使用方法

数据加载器,结合了数据集和取样器,并且可以提供多个线程处理数据集。在训练模型时使用到此函数,用来把训练数据分成多个小组,此函数每次抛出一组数据。直至把所有的数据都抛出。就是做一个数据的初始化。生成迭代数据非常方便,请看如下示例:""" 批训练,把数据变成一小批一小批数据进行训练。 DataLoader就是用来包装所使用的数据,每次抛出一批数据"""imp...