測驗了 nn.CrossEntropyLoss的檔案示例。它似乎不起作用。
loss = nn.CrossEntropyLoss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5).softmax(dim=1)
output = loss(input, target)
print(output)
錯誤:
RuntimeError: 1D target tensor expected, multi-target not supported
uj5u.com熱心網友回復:
您可能需要升級您的手電筒版本。
pip install torch --upgrade
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/472464.html
上一篇:有沒有辦法運行這個腳本?
下一篇:多類的SHAP值解釋
