为什么这个语义分割网络在pytorch中没有softmax分类层?

您使用相当复杂的代码进行培训/推理。但是如果你稍微挖掘一下,你会发现损失函数已经实现了。 here 你的模型实际上是用 cross_entropy 损失。看医生:

该标准将对数软最大值和nll损失结合在一个函数中。

对于数值稳定性而言,最好是将SoftMax“吸收”到损失函数中,而不是通过模型进行明确计算。
这是一个非常常见的做法,让模型输出“原始”预测(也称为“逻辑”),然后让损失(也称为标准)在内部执行SoftMax。 如果您真的需要概率,您可以在部署模型时在顶部添加一个softmax。