RuntimeError: Could not infer dtype of NoneType
RuntimeError: Could not infer dtype of NoneType出错原因dataloader在dataset中一个一个读取数据的时候遇到了脏数据(空值之类的),导致在网络推理的时候出现了错误。解决方法方法一:跳过该数据在enumerate一个一个读取数据的时候判断读到的数据是否是空的,是的话当前循环continue方法二:使用collate_fn函数https://b
·
RuntimeError: Could not infer dtype of NoneType
出错原因
dataloader在dataset中一个一个读取数据的时候遇到了脏数据(空值之类的),导致在网络推理的时候出现了错误。
解决方法
方法一:跳过该数据
在数据放入dataset中之前判断image 和 label是否为空,如果为空则跳过该数据,不把他存入dataset
方法二:使用collate_fn函数
https://blog.csdn.net/guyuealian/article/details/91129367
参考链接
别人的解决方案:
https://www.yht7.com/news/135031
判断数据是否为空用pandas.isnull(obj),记得import pandas
https://book.itheima.net/course/221/1270308811000782849/1271373716495802370
关于PyTorch读取数据的dataloader、dataset具体情况:
https://pytorch.org/tutorials/beginner/basics/data_tutorial.html
更多推荐
已为社区贡献2条内容
所有评论(0)