Linear(),softmax(),CrossEntropyLoss()的一些理解

    技术2025-02-26  11

    pytorch的图像分类问题 前一段时间突然迷了,最后的全连接层(FC)输出的维度对应的分类数目(Class_num),这点是没错的,但是交叉熵损失函数公式里边是有预测概率的,必须由softmax得到,但是程序里没有看到softmax(),但是很多都会说到softmax层。后来经过查阅发现,解决办法在交叉熵损失函数里边。 是由交叉熵损失函数的结构解决。

    交叉熵损失函数: 链接: https://blog.csdn.net/c2250645962/article/details/106014693/?utm_medium=distribute.pc_relevant.none-task-blog-baidujs-1.

    Processed: 0.008, SQL: 9