Conv1d参数与输入数据均无nan,结果出现nan的问题解决
解决方法:
极有可能是CUDA和pytorch版本适配的问题。博主在复现LaneGCN时,由于我们学校的服务器上的CUDA版本是11.1,与GitHub上官方代码给的pytorch的版本不一致,所以导致了这一令人完全摸不着头脑的问题。不过这也给我积累了一些经验,就是当你认为输入没有问题,模型没有问题(结构和参数没有问题),那就多半是一些不适配的问题(例如,CUDA和pytorch)。 由于服务器上的CUDA好像不能更改(似乎是可以的,但是挺麻烦),因此,我使用conda创建一个新的虚拟环境,在其中,使用pytorch官方的代码进行对应版本pytorch的配置:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia
由于博主还是使用pytorch进行神经网络训练的小白,这篇博客是博主含泪用
版权声明:
作者:lichengxin
链接:https://www.techfm.club/p/8487.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。
THE END
二维码
共有 0 条评论