首页 Paddle框架 帖子详情
Dataloader返回LoDTensor
收藏
快速回复
Paddle框架 问答模型训练 495 1
Dataloader返回LoDTensor
收藏
快速回复
Paddle框架 问答模型训练 495 1

在2.1版本中,按照API文件描述,paddle.io.DataLoader返回的是Tensor类型的变量,实际运行中,返回的是LoDTensor类型,代码及运行结果如下,恳请指教:

train_data = paddle.text.datasets.UCIHousing(mode='train')

train_loader = paddle.io.DataLoader(train_data, batch_size=batch_size, shuffle=True)

for data, label in train_loader():
    print(type(data))

运行耗时: 28毫秒



0
收藏
回复
全部评论(1)
时间顺序
郭天伟
#2 回复于2021-05

重启notebook之后运行正常……

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