CNN中分类头和回归头的损失权重

时间:2018-11-08 03:07:43

标签: classification regression conv-neural-network loss

我有一个CNN,它的分类头有2个输出,使用分类交叉熵;而回归头有2个输出,使用均方误差作为损失。

CNN的损失是分类头和回归头的损失的线性组合:

cnn_loss = W_class *损失分类+ W_regression *损失回归

是否有一种很好的方法来确定W_class和W_regression的值或常用值?

0 个答案:

没有答案