- 博客(1)
- 收藏
- 关注
RSS订阅原创 pytorch和Keras中的softmax分类
pytorch中的CrossEntropyLoss自带了Softmax激活运算，因此在模型中最后一层只是一个全连接线性层Linear，后面不要再激活。 Keras中的categorical_crossentropy只计算交叉熵，因此模型最后一层Dense需要用softmax激活。 ...
2021-08-29 23:01:09
251
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人