项目场景:
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!
我赶紧改了
结果终于正确了!!!