cuda error invalid argument

待我称王封你为后i 2022-10-09 03:22 334阅读 0赞

用pytorch训练时报的错

在loss.backward时报错

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

网上其他答案:

问题:
RuntimeError: CUDA error: invalid configuration argument

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

发表评论

表情:
评论列表 (有 0 条评论,334人围观)

还没有评论,来说两句吧...

相关阅读