RuntimeError: Caught RuntimeError in DataLoader worker process 3.RuntimeError: torch.cat(): input ty
Dataloader的自定义dataset多线程压缩数据集出现问题RuntimeError: Caught RuntimeError in DataLoader worker process 3.RuntimeError: torch.cat(): input types can't be cast to the desired output type Long.
·
项目场景:
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!
我赶紧改了
结果终于正确了!!!
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)