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