我对PyTorch dataloader里的shuffle=True的理解
对shuffle=True的理解:
之前不了解shuffle的实际效果,假设有数据a,b,c,d,不知道batch_size=2后打乱,具体是如下哪一种情况:
1.先按顺序取batch,对batch内打乱,即先取a,b,a,b进行打乱;
2.先打乱,再取batch。
证明是第二种
shuffle (bool, optional): set to ``True`` to have the data reshuffled
at every epoch (default: ``False``).
if shuffle:
sampler = RandomSampler(dataset) #此时得到的是索引
补充:简单测试一下pytorch dataloader里的shuffle=True是如何工作的
看代码吧~
import sys
import torch
import random
import argparse
import numpy as np
import pandas as pd
import torch.nn as nn
from torch.nn import functional as F
from torch.optim import lr_scheduler
from torchvision import datasets, transforms
from torch.utils.data import TensorDataset, DataLoader, Dataset
class DealDataset(Dataset):
def __init__(self):
xy = np.loadtxt(open('./iris.csv','rb'), delimiter=',', dtype=np.float32)
#data = pd.read_csv("iris.csv",header=None)
#xy = data.values
self.x_data = torch.from_numpy(xy[:, 0:-1])
self.y_data = torch.from_numpy(xy[:, [-1]])
self.len = xy.shape[0]
def __getitem__(self, index):
return self.x_data[index], self.y_data[index]
def __len__(self):
return self.len
dealDataset = DealDataset()
train_loader2 = DataLoader(dataset=dealDataset,
batch_size=2,
shuffle=True)
#print(dealDataset.x_data)
for i, data in enumerate(train_loader2):
inputs, labels = data
#inputs, labels = Variable(inputs), Variable(labels)
print(inputs)
#print("epoch:", epoch, "的第" , i, "个inputs", inputs.data.size(), "labels", labels.data.size())
简易数据集
shuffle之后的结果,每次都是随机打乱,然后分成大小为n的若干个mini-batch.
(资源库 www.zyku.net)
原文链接:https://blog.csdn.net/qq_35248792/article/details/109510917
上一篇:浅谈pytorch中为什么要用 zero_grad() 将梯度清零
栏 目:Python教程
下一篇:PyTorch梯度裁剪避免训练loss nan的操作
本文标题:我对PyTorch dataloader里的shuffle=True的理解
本文地址:https://www.zyku.net/python/9584.html
您可能感兴趣的文章
- 04-09使用pytorch实现线性回归
- 02-10小米11接入点APN更改方法
- 02-20Sublime Text 3中文优化版相关配置教
- 09-25安装帝国CMS遇到“修改php.ini,将:short
- 02-19discuzX3.1门户keyword和description
- 03-31华为手机停车位置功能开启方法
- 01-12大白答题软件-大白答题软件应用软件功
- 01-14鹫一健康-鹫一健康应用软件功能介绍
- 03-25华为nova8关闭触屏亮屏功能方法
- 09-19人民日报在哪里看中小学生公益直播课
- 01-13Yorbit记录生活-Yorbit记录生活应用软
- 04-15荣耀v40轻奢版智慧语音唤醒教程
- 01-12时光相册你的名字滤镜app-时光相册你
- 12-25氧气动态壁纸-氧气动态壁纸应用软件功
- 12-30WOLB-WOLB应用软件功能介绍
- 11-30苹果手机怎样取消访问限制
- 01-14云上中文-云上中文应用软件功能介绍
- 03-03苹果手机设置轻点两下截图教程
- 10-28苹果13如何设置屏幕亮度
- 10-12ipados15.0.2更新内容分享
最近更新
阅读排行
猜你喜欢
- 09-26小米耳机air2se如何再次配对
- 10-21华为耳机怎么重新匹配手机
- 01-12拼搏-拼搏应用软件功能介绍
- 12-29ZTISmart-ZTISmart应用软件功能介绍
- 11-12vivoX70Pro+丢了怎么找回
- 08-05宝塔Linux面板定时备份网站数据至七牛
- 02-28CentOS 6.4下安装MySQL后重置root密码
- 12-21荣耀畅玩30plus如何设置息屏显示
- 04-27OPPO手机剪辑视频教程
- 09-18苹果耳机电量插件添加教程介绍