字典定义 "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改为其他类型。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)