center loss 与 softmax 交叉熵的监督loss是怎么实现的
可以看出,是计算了center loss 以及 softmax ,然后将两个loss进行相加。
center loss 的输入是倒数第二层全连接层的输出维度是[128,2]
features,输出为 center loss
softmax 交叉熵函数 的输入为model的输出,即全连接的输出 维度是[128,10]
最后,将softmax的结果与center loss乘以权重之后的结果相加。
center 的更新是在batch size里