weixin_42486139的博客

center loss来自ECCV2016的一篇论文：A Discriminative Feature Learning Approach for Deep Face Recognition 公式： 其中， x指的是特征，cyi指的是第yi个类别的中心，c会随着模型训练更新，类中心数=类别数； m表示mini-batch的大小,因此这个公式就是希望一个batch中的每个样本的feature离feature 的中心的距离的平方和要越小越好，也就是类内距离要越小越好。 实现代码： cl...

if len(gpu_ids) ==1: t.save(net.module.state_dict(), "model.pth") else: t.save(net.state_dict(), "model.pth")