ArcFace/SphereFace Loss
Compare changes
- Tiago de Freitas Pereira authored
+ 1
− 1
@@ -166,7 +166,7 @@ class ModifiedLogitsHead(tf.keras.layers.Layer):
Implemented 2 losses and 1 layer from the arcface paper: https://arxiv.org/abs/1801.07698
Implemented 1 loss and 1 layer from the sphere face paper: https://arxiv.org/pdf/1704.08063.pdf
Implemented a keras model that does the validation in the test step