摘要: 下面注释insightface中 大规模人脸分类层实现细节 完整分类代码放到页面末尾 大规模人脸分类的基本思想是,分类层可线性拆分,可将分类层均匀拆分到N块GPU上。分类类别数量增大,仅需增加GPU的数量即可,单卡存放部分的分类层参数量可不变。 (1)下面代码就是实现分类层如何拆分。如果分类类别数量 阅读全文
posted @ 2021-08-29 18:36 星辰大海,绿色星球 阅读(294) 评论(0) 推荐(0) 编辑
摘要: 下面主要注释了百度开源人脸大规模分类训练PLSC代码细节 (1)下面代码softmax_with_cross_entropy之前部分主要是使用allgather将每个gpu上计算的特征聚合在一起,每块gpu上都存放了相同的全体特征。 def softmax_classify(self, x, lab 阅读全文
posted @ 2021-08-29 18:36 星辰大海,绿色星球 阅读(206) 评论(0) 推荐(0) 编辑