模型训练五部曲：数据、模型、损失函数、优化器、迭代训练
一 数据
1.Dataloader
torch.utils.data.Dataloader
功能：构建可迭代带数据装载器
dataset:Dataset类，决定数据从哪读取及如何读取
batchsize:批大小
num_works：是否多进程读取数据
shuffle：每个epoch是否乱
drop_last:当样本数不能被batchsize整除时，是否舍弃最后一批数据
相关文章：
相似解决方案
热门标签