insightface
insightface copied to clipboard
Why embedding needs gradient in PFC
Line188 partial_fc.py and Line139 in partial_fc_v2.py is
_list_embeddings = AllGather(local_embeddings, *_gather_embeddings)
Why embeddings needs gradient across multi Gpus?