简单粗暴PyTorch之Dataloader与Dataset(数据读取的核心)

Dataloader与Dataset一、Dataloader二、Dataset机器学习模型训练步骤由数据、模型、损失函数、优化器、迭代训练五部分组成。数据部分由数据收集、数据划分、数据读取、数据预处理四部分组成。Dataloader与Dataset是Pytorch数据读取的核心一、Dataloadertorch.utils.data.DataLoader()功能:构建可迭代的数据装载器• dataset: Dataset类,决定数据从哪读取及如何读取• batchsize : 批大小•
摘要由CSDN通过智能技术生成

Dataloader与Dataset


注:来自 深度之眼Pytorch框架班
机器学习模型训练步骤由数据、模型、损失函数、优化器、迭代训练五部分组成。

在这里插入图片描述
数据部分由数据收集、数据划分、数据读取、数据预处理四部分组成。
Dataloader与Dataset是Pytorch数据读取的核心

一、Dataloader

torch.utils.data.DataLoader()

功能:构建可迭代的数据装载器
• dataset: Dataset类,决定数据从哪读取及如何读取
• batchsize : 批大小
• num_works: 是否多进程读取数据
• shuffle: 每个epoch是否乱序
• drop_last:当样本数不能被batchsize整 除时,是否舍弃最后一批数据

在这里插入图片描述

Epoch: 所有训练样本都已输入到模型中,称为一个Epoch
Iteration:一批样本输入到模型中,称之为一个Iteration
Batchsize:批大小,决定一个Epoch有多少个Iteration
样本总数࿱

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值