首页 Paddle框架 帖子详情
使用DataLoader的技巧
收藏
快速回复
Paddle框架 文章深度学习炼丹技巧 613 1
使用DataLoader的技巧
收藏
快速回复
Paddle框架 文章深度学习炼丹技巧 613 1

使用 DataLoader 读取图片数据时,注意在 DataSet 的 __getitem__() 方法里返回图片的label时,要将label的数据类型设为整型,不要设为和图片数据一样的float32,否则可能报错。

0
收藏
回复
全部评论(1)
时间顺序
AIStudio810258
#2 回复于2021-02

因为使用DataLoader返回的数据会被自动打包成 paddle的tensor,可能对数据类型有要求

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