1. 首页
  2. 数据库
  3. 其它
  4. PyTorch学习笔记(4)Dataloader与Dataset(1)

PyTorch学习笔记(4)Dataloader与Dataset(1)

上传者: 2021-01-17 04:44:05上传 PDF文件 28.67KB 热度 16次
torch.utils.data.DataLoader dataset Dataset类 决定数据从哪读取及如何读取 batchsize 批大小 num_works 是否多进程读取数据 shuffle 每个epoch 是否乱序 drop_last 当样本数不能被batchsize整除时,是否舍弃最后一批数据 Epoch 所有训练样本都已输入到模型中,成为一个Epoch Iteration 一批样本输入到模型中,称之为一个Iteration Batchsize 批大小 决定一个Epoch有多少个Iteration 例如 样本总数80 Batchsize 8 1Epoch = 10 Iterat
用户评论