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

本文介绍了PyTorch中数据读取的关键组件Dataloader和Dataset。Dataloader用于构建可迭代的数据加载器,涉及参数如batch size、多进程读取和数据乱序。Dataset是数据源抽象类,需自定义并实现getitem()方法来获取单个样本。理解这两个概念对于有效地处理数据集和训练模型至关重要。
摘要由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
样本总数࿱

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值