diff --git a/test/test_losses.py b/test/test_losses.py index 8b55729..40c1a5a 100644 --- a/test/test_losses.py +++ b/test/test_losses.py @@ -16,7 +16,7 @@ def arcface_loss_val(embedding, labels, weights, out_num, s=64., m=0.5): ''' cos_m = math.cos(m) sin_m = math.sin(m) - mm = sin_m * m * s + mm = sin_m * m # issue 1 threshold = math.cos(math.pi - m) with tf.variable_scope('arcface_loss'): # inputs and weights norm @@ -134,4 +134,4 @@ def mxnet_arcface_val(embedding, gt_label, weights): print(len(result2[0])) print(type(result1)) print(type(result2[0].asnumpy())) - print(np.mean(result1 - result2[0].asnumpy())) # 1.26362e-07 \ No newline at end of file + print(np.mean(result1 - result2[0].asnumpy())) # 1.26362e-07