人脸识别
文章平均质量分 82
彩虹宇
这个作者很懒,什么都没留下…
展开
-
使用torch.autograd.function解决dist.all_gather不能反向传播问题
最近在用mmcv复现Partial FC模型,看到源码中,有单独写的前向反向传播,甚是疑惑~源码: # Features all-gathertotal_features = torch.zeros(features.size()[0] * cfg.world_size, cfg.embedding_size, device=local_rank)dist.all_gather(lis原创 2021-06-30 16:13:06 · 1876 阅读 · 1 评论 -
人脸识别论文:Partial FC: Training 10 Million Identities on a Single Machine
1. 背景介绍在当前的分类任务研究中,大多数损失函数都是基于Softmax Loss或者其变体(如NormFace、ArcFace、Circle Loss、AdaCos等改进),训练效果也很好,但是当数据集的类别数量较大时,对于GPU的占用也会特别的大。训练一个百万甚至千万级别的人脸数据集,基于Softmax Loss或者其改进形式作为损失的模型是非常耗时的。DeepInsight发布的Partial-FC人脸识别模型,对传统的softmax层进行了优化,一定程度上解决了大规模人脸识别模型训练中,对GP原创 2021-03-31 14:43:14 · 1541 阅读 · 0 评论