我正在对我的数据集进行微调,它有多个标签。
我想将分类层的“分数”转换为概率,并使用这些概率来计算训练时的损失。
你能给出一个这样的示例代码吗?我可以这样使用吗:
P = net.forward(x)
p = torch.nn.functional.softmax(P, dim=1)
loss = torch.nn.functional.cross_entropy(P, y)
我不清楚这是不是正确的方法,因为我将概率作为交叉损失的输入。
https://stackoverflow.com/questions/51291353
复制相似问题