Pytorch交叉熵损失函数CrossEntropyLoss报错解决办法

在pytorch的官方论坛里有一我的也遇到了一样的问题,他把nn.CrossEntropyLoss()换成了nn.MultiLabelSoftMarginLoss()就再也不报错了。可是通过实验发现,这个损失函数的效果很是差,远不如MSELoss。web 最终,我找到了一篇运用交叉熵损失函数的多分类代码一步步检查发现了报错的缘由:svg 在多分类问题中,当损失函数为nn.CrossEntropyL
相关文章
相关标签/搜索