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

cuda error invalid argument

2021-06-23 23:45:30 Wangqi

use pytorch The training times is wrong

stay loss.backward Times wrong

It should be loss There's an outlier .

Other answers on the Internet :

problem :
RuntimeError: CUDA error: invalid configuration argument

reason : The parameter is too large , Too much memory
resolvent : Check whether the previous code and network design are reasonable , Is there anything wrong . My mistake is that I forgot to add pooling layer before , Cause the data to be too big , To report a mistake .

版权声明
本文为[Wangqi]所创,转载请带上原文链接,感谢
https://chowdera.com/2021/06/20210623234449690Q.html

随机推荐