怎么解決pytorch損失反向傳播后梯度為none的問題

pytorch在進行有損失的反向傳播的時候，有時候會出現(xiàn)梯度為none的情況，那么這種情況要如何解決呢？來看看小編是怎么做的：

錯誤代碼：輸出grad為none

a = torch.ones((2, 2), requires_grad=True).to(device)
b = a.sum()
b.backward()
print(a.grad)

由于.to(device)是一次操作，此時的a已經(jīng)不是葉子節(jié)點了

修改后的代碼為：

a = torch.ones((2, 2), requires_grad=True)
c = a.to(device)

b = c.sum()
b.backward()
print(a.grad)

類似錯誤：

self.miu = torch.nn.Parameter(torch.ones(self.dimensional)) * 0.01

應該為

self.miu = torch.nn.Parameter(torch.ones(self.dimensional) * 0.01)

補充：pytorch梯度返回none的bug

pytorch1.4.0如果使用了view方法，reshape方法

tensor即使設置了requires_grad，反向傳播之后， x返回沒有grad梯度，為none

不知道其他版本有無此bug

補充：PyTorch中梯度反向傳播的注意點

在一個迭代循環(huán)中

optimizer.zero_grad()語句的位置比較隨意，只要放在loss.backward()前面即可，它的作用是將梯度歸零，否則會在每一個迭代中進行累加，

loss.backward()的作用是反向傳播，計算梯度，optimizer.step()的功能是優(yōu)化器自動完成參數(shù)的更新。

optimizer.zero_grad()
loss.backward()
optimizer.step()

以上就是怎么解決pytorch損失反向傳播后梯度為none的問題的全部內(nèi)容，希望能給大家一個參考，也希望大家多多支持W3Cschool。