问如何使用pytorch从Resnet获取概率？
EN

Stack Overflow用户

提问于 2018-07-12 01:36:43

回答 1查看 7.1K关注 0票数 4

我正在对我的数据集进行微调，它有多个标签。

我想将分类层的“分数”转换为概率，并使用这些概率来计算训练时的损失。

你能给出一个这样的示例代码吗？我可以这样使用吗：

       P = net.forward(x)
       p = torch.nn.functional.softmax(P, dim=1)
       loss = torch.nn.functional.cross_entropy(P, y)

我不清楚这是不是正确的方法，因为我将概率作为交叉损失的输入。

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/51291353

复制

相似问题

问如何使用pytorch从Resnet获取概率？EN