网站介绍:文章浏览阅读4.5k次,点赞28次,收藏35次。起因最近在使用pytorch训练模型时,出现了loss下降非常缓慢的问题,想了很久都没有发现问题出在哪里。_使用torch的crossentropyloss需不需要softmax
- 链接地址:https://blog.csdn.net/weixin_44607838/article/details/118119326
- 链接标题:Pytorch 中使用nn.CrossEntropyLoss的注意点(不需要额外的softmax)_使用torch的crossentropyloss需不需要softmax-CSDN博客
- 所属网站:blog.csdn.net
- 被收藏次数:5328
- 网站标签:使用torch的crossentropyloss需不需要softmax