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

模型训练五部曲：数据、模型、损失函数、优化器、迭代训练 一 数据 1.Dataloader torch.utils.data.Dataloader 功能：构建可迭代带数据装载器 dataset:Dataset类，决定数据从哪读取及如何读取 batchsize:批大小 num_works：是否多进程读取数据 shuffle：每个epoch是否乱 drop_last:当样本数不能被batchsize整

>>阅读原文<<