其他分享
首页 > 其他分享> > Pytorch——Dataset&Dataloader

Pytorch——Dataset&Dataloader

作者:互联网

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

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

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

 

标签:训练,包装,Dataloader,Dataset,Pytorch,数据
来源: https://www.cnblogs.com/brady12/p/16634408.html