Pytorch入门---学习笔记五---Dataloader与Dataset

模型训练五部曲:数据、模型、损失函数、优化器、迭代训练
一 数据
Pytorch入门---学习笔记五---Dataloader与Dataset
1.Dataloader
torch.utils.data.Dataloader
功能:构建可迭代带数据装载器
dataset:Dataset类,决定数据从哪读取及如何读取
batchsize:批大小
num_works:是否多进程读取数据
shuffle:每个epoch是否乱
drop_last:当样本数不能被batchsize整除时,是否舍弃最后一批数据
Pytorch入门---学习笔记五---Dataloader与Dataset

iteration=epoch/batchsize

Pytorch入门---学习笔记五---Dataloader与Dataset