diff --git a/bob/learn/tensorflow/losses/center_loss.py b/bob/learn/tensorflow/losses/center_loss.py index a6da47f618c0ba87ef7e41e44fdb2af58a022916..19196df105d6ff52114c42439cb09228c887b9e3 100644 --- a/bob/learn/tensorflow/losses/center_loss.py +++ b/bob/learn/tensorflow/losses/center_loss.py @@ -6,7 +6,7 @@ class CenterLossLayer(tf.keras.layers.Layer): Attributes ---------- - centers: + centers The variable that keeps track of centers. n_classes : int