Hi, Thank you for your great job! I have a question, how to implement CenterLoss using Cosine distance rather than Euclidean distance? Thank you!