首页 Paddle框架 帖子详情
使用API Dataset、DataLoader时的WARNING有误。paddle框架为2.2版本
收藏
快速回复
Paddle框架 问答模型训练 693 4
使用API Dataset、DataLoader时的WARNING有误。paddle框架为2.2版本
收藏
快速回复
Paddle框架 问答模型训练 693 4

WARNING: Detect dataset only contains single fileds, return format changed since Paddle 2.1. In Paddle <= 2.0, DataLoader add a list surround output data(e.g. return [data]), and in Paddle >= 2.1, DataLoader return the single filed directly (e.g. return data).

paddle已经是2.2版本的, 但是__getitem__返回数据不加 [] 就会WARNING

啥时候修复错误

快速复现该WARNING:

import numpy as np
import paddle
from paddle.io import DataLoader

# define a random dataset
class Dataset(paddle.io.Dataset):
    def __init__(self, images):
        self.images = images[:]

    def __getitem__(self, index):
        return self.images[index].astype('float32')
#         return [self.images[index].astype('float32')]

    def __len__(self):
        return len(self.images)

images = paddle.rand((100,224,224))
i_query_data = Dataset( images)
loader = DataLoader(i_query_data,
                    batch_size=64,
                    shuffle=True,
                    drop_last=True,
                    num_workers=0)


for i,image in enumerate(loader()):
    print(i,image)
    break

 

0
收藏
回复
全部评论(4)
时间顺序
三岁
#2 回复于2021-11

可以去GitHub上提一个issue哦

 

0
回复
宇宙物语
#3 回复于2021-11
三岁 #2
可以去GitHub上提一个issue哦  

忘了。我现在就去反馈

0
回复
三岁
#4 回复于2021-11
忘了。我现在就去反馈

好哒~~~

0
回复
十进制到二进制
#5 回复于2021-11

2.1.2 无该错误。

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