pytorch中的CrossEntropyLoss自带了Softmax激活运算,因此在模型中最后一层只是一个全连接线性层Linear,后面不要再激活。
Keras中的categorical_crossentropy只计算交叉熵,因此模型最后一层Dense需要用softmax激活。
pytorch和Keras中的softmax分类
最新推荐文章于 2023-08-24 01:09:32 发布
pytorch中的CrossEntropyLoss自带了Softmax激活运算,因此在模型中最后一层只是一个全连接线性层Linear,后面不要再激活。
Keras中的categorical_crossentropy只计算交叉熵,因此模型最后一层Dense需要用softmax激活。