AttributeError:“ Tensor”没有属性:“ backwards”

时间:2018-10-15 12:44:52

标签: pytorch loss-function

我要向那些犯了与我相同的错误的人发布此问题。尝试计算渐变时出现此错误:

criterion = torch.nn.CrossEntropyLoss()
loss = criterion(y_hat, y_truth)
loss.backwards()

1 个答案:

答案 0 :(得分:5)

它是loss.backward(),而不是loss.backwards()