2021年10月5日
人脸大规模分类—计算过程
摘要： 这里简要总结一下，人脸大规模分类的实现过程 （1）拆分分类权重：分类权重$W\in R^{n\times d}$拆分为K份，每个gpu存储其中一份分类权重 （2）求logit值：每个gpu上的特征feature，及其label，经过allgather操作汇聚并广播到每个GPU上，然后和当前gpu上的 阅读全文
posted @ 2021-10-05 09:22 星辰大海,绿色星球 阅读(64) 评论(0) 推荐(0) 编辑