- 博客(2)
- 收藏
- 关注
原创 【代码小坑】梯度回传为0
记录一下训练过程中遇到的问题,由于这个问题我花了很长时间才解决,所以值得记录。先给出结论:tensor转换成numpy后会丢失梯度,导致回传出现问题由于原代码不容易理解,简单用个例子说明import torchfrom torch.autograd import Variableinput=Variable(torch.tensor([1,2,3.]))input.requires_grad=Trueinput_numpy=input.detach().numpy()x=input
2022-05-10 11:16:48 1460 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人