Jac*_*ern 2 pytorch loss-function
我要向那些犯了与我相同的错误的人发布此问题。尝试计算渐变时出现此错误:
criterion = torch.nn.CrossEntropyLoss()
loss = criterion(y_hat, y_truth)
loss.backwards()
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
856 次 |
| 最近记录: |