How to implement arcface without Partial FC with pytorch?

I find that it is forced to use partial fc in arcface_torch, but I just want to use arcface. Is there any offical implement with pytorch? Or need I implement it by myself (any advice to do this, I'm worried about some tricks I didn't think of...)? Thanks!

Start with the arcface_torch implementation and its Partial FC usage, then determine whether the repository contains an official PyTorch path that avoids Partial FC. If no such path is identified, document the required implementation scope and the training details that must be preserved; the issue currently names no files or tests.

Written by the indexing model from the issue text.