你的位置:首页 > 信息动态 > 新闻中心
信息动态
联系我们

Pytorch——DatasetDataloader

2022/8/28 23:22:50

在利用 Pytorch 进行深度学习的训练时需要将数据进行打包,这就是 Dataset 与 Dataloader 的作用。 

Dataset 将数据进行包装,Dataloader 迭代包装好的数据并输出每次训练所需要的矩阵。 

官网教程: Datasets & DataLoaders — PyTorch Tutorials 1.12.1+cu102 documentation