arminbiglari

Hello, i resolved the problem by changing the code to ``` model = MobileFaceNet(embedding_size).to(device) head = Arcface(embedding_size=embedding_size, classnum=6056).to(device) head_params = [param for name, param in head.named_parameters()] # Manually create parameter...