【pytorch字典TypeError: batch must contain tensors, numbers, dicts or lists; found <class ‘NoneType‘>】

【pytorch字典TypeError: batch must contain tensors, numbers, dicts or lists; found <class ‘NoneType‘>】,第1张

【pytorch字典TypeError: batch must contain tensors, numbers, dicts or lists; found <class ‘NoneType‘>】
字典定义
 "name": null, 
 "shape": "258", 
 "list": [], 

使用DataLoader从json文件读取:

from torch.utils.data import Dataset, DataLoader
dataProvider = DataProvider()
train_loader = DataLoader(dataProvider)

报错

    raise TypeError((error_msg.format(type(batch[0]))))
TypeError: batch must contain tensors, numbers, dicts or lists; found 

原因是json文件中的null虽然能解析成python的字典,但是Pytorch的DataLoader不支持。将null改为其他类型。

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5711645.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存