Pytorch 中使用nn.CrossEntropyLoss的注意点(不需要额外的softmax)_使用torch的crossentropyloss需不需要softmax-CSDN博客

网站介绍:文章浏览阅读4.5k次,点赞28次,收藏35次。起因最近在使用pytorch训练模型时,出现了loss下降非常缓慢的问题,想了很久都没有发现问题出在哪里。_使用torch的crossentropyloss需不需要softmax