在运行神经网络求梯度的时候遇到了这样的问题。
发现出错在
原因:这里的loss默认的requires_grad是False,因此在backward()处不会计算梯度,导致出错
方法:将loss的requires_grad属性设为True,如下所示:
在运行神经网络求梯度的时候遇到了这样的问题。
发现出错在
原因:这里的loss默认的requires_grad是False,因此在backward()处不会计算梯度,导致出错
方法:将loss的requires_grad属性设为True,如下所示: