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