项目场景:

RuntimeError: Caught RuntimeError in DataLoader worker process 3.RuntimeError: torch.cat(): input types can’t be cast to the desired output type Long.

遇到一个很搞笑的bug,在利用torch的DataLoader打包数据集,自己写了dataset函数,但是在把数据放进去的时候系统提示多线程报告错误,另一个错误是说数据拼接打包出现错误,

问题描述

在这里插入图片描述
这里描述项目中遇到的问题:`我记得之前也遇到过,但没解决,网上搜的办法只告诉我数据维度不对,但是我输出了数据维度是没错的,我仔细阅读报错,才发现是数据类型问题。


原因分析::

读入的数据默认为int,但其他的数据是float,转换为tensor读取的时候出现了问题。


解决方案:

把数据手动改为float!

在这里插入图片描述我赶紧改了
在这里插入图片描述结果终于正确了!!!

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐