当前位置:网站首页>cuda error invalid argument

cuda error invalid argument

2021-06-23 23:45:04 网奇

用pytorch训练时报的错

在loss.backward时报错

估计应该是loss出现了异常值。

网上其他答案:

问题:
RuntimeError: CUDA error: invalid configuration argument

原因:参数过大,内存需求过大
解决方法:查看之前的代码及网络设计是否合理,是否有错误的地方。本人错误在于之前忘记了加池化层,导致数据过大,从而报错。

版权声明
本文为[网奇]所创,转载请带上原文链接,感谢
https://blog.csdn.net/jacke121/article/details/118157168

随机推荐