Corrected loss function name

YesterdayisTuesday · Jan 19, 2017 · 5df52a3 · 5df52a3
1 parent f5ba23e
commit 5df52a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/center_loss_test.py b/test/center_loss_test.py
@@ -41,7 +41,7 @@ def testCenterLoss(self):
             labels = tf.placeholder(tf.int32, shape=(batch_size,), name='labels')
 
             # Define center loss
-            center_loss, centers = facenet.center_loss_new(features, labels, alfa, nrof_classes)
+            center_loss, centers = facenet.center_loss(features, labels, alfa, nrof_classes)
 
             label_to_center = np.array( [ 
                  [-3,-3],  [-3,-1],  [-3,1],  [-3,3],