DataLoader加载MNIST数据集后,label值出现错误,什么原因?
收藏
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
收藏
请登录后评论
收到发多少