【发布时间】:2019-03-19 21:40:43
【问题描述】:
我为那些犯了同样错误的人发布这个问题。尝试计算梯度时出现此错误:
criterion = torch.nn.CrossEntropyLoss()
loss = criterion(y_hat, y_truth)
loss.backwards()
【问题讨论】:
标签: pytorch loss-function
我为那些犯了同样错误的人发布这个问题。尝试计算梯度时出现此错误:
criterion = torch.nn.CrossEntropyLoss()
loss = criterion(y_hat, y_truth)
loss.backwards()
【问题讨论】:
标签: pytorch loss-function
这是loss.backward(),而不是loss.backwards()。
【讨论】: