用pytorch训练时报的错
在loss.backward时报错
估计应该是loss出现了异常值。
网上其他答案:
问题:
RuntimeError: CUDA error: invalid configuration argument
原因:参数过大,内存需求过大
解决方法:查看之前的代码及网络设计是否合理,是否有错误的地方。本人错误在于之前忘记了加池化层,导致数据过大,从而报错。
用pytorch训练时报的错
在loss.backward时报错
估计应该是loss出现了异常值。
网上其他答案:
问题:
RuntimeError: CUDA error: invalid configuration argument
原因:参数过大,内存需求过大
解决方法:查看之前的代码及网络设计是否合理,是否有错误的地方。本人错误在于之前忘记了加池化层,导致数据过大,从而报错。