标签: pytorch loss-function
我要向那些犯了与我相同的错误的人发布此问题。尝试计算渐变时出现此错误:
criterion = torch.nn.CrossEntropyLoss() loss = criterion(y_hat, y_truth) loss.backwards()
答案 0 :(得分:5)
它是loss.backward(),而不是loss.backwards()。
loss.backward()
loss.backwards()