首页 Paddle框架 帖子详情
DataLoader加载MNIST数据集后,label值出现错误,什么原因?
收藏
快速回复
Paddle框架 问答模型训练 813 1
DataLoader加载MNIST数据集后,label值出现错误,什么原因?
收藏
快速回复
Paddle框架 问答模型训练 813 1

PaddlePaddle版本:2.6
Python版本: 3.9.0
系统:Windows 11

import paddle
import paddle.vision.transforms as T
from paddle.vision.datasets  import MNIST
from paddle.io import DataLoader


paddle.device.set_device("gpu:0")
mnist = MNIST(mode="train", transform=T.ToTensor())
for i in range(5):
    img, label = mnist[i]
    print(img.shape, label)


train_loader = DataLoader(mnist, batch_size=5, shuffle=False)
for idx,data in enumerate(train_loader()):
    img= data[0]
    label= data[1]
    if idx == 0:
        print("DataLoader img.shape:{},\nDataLoader label:{}".format(img.shape,label))
        break

CUDA版本是11.6;cuDNN版本是8.4.0。

程序运行的结果:

 问题:DataLoader加载MNIST数据集后,label的值应当也是[[5],[0],[4],[1],[9]],因为Dataloader加载时设置的是不打乱顺序。但运行结果label值很奇怪,好像是随机的值。试了CUDA 11.7 + cuDNN 8.4.1,结果是同样的问题。请问是什么问题?谢谢。

 

0
收藏
回复
全部评论(1)
时间顺序
j
jzhang533
#2 回复于2024-12

收到发多少

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户