nitpick

bob/learn/tensorflow/loss/BaseLoss.py

@@ -4,7 +4,7 @@
 
 import logging
 import tensorflow as tf
-logger = logging.getLogger("bob.learn.tensorflow")
+logger = logging.getLogger(__name__)
 
 slim = tf.contrib.slim
 

bob/learn/tensorflow/loss/ContrastiveLoss.py

@@ -4,9 +4,7 @@
 
 import logging
 import tensorflow as tf
-from bob.learn.tensorflow.utils import (
-    compute_euclidean_distance,
-)
+from bob.learn.tensorflow.utils import compute_euclidean_distance
 
 logger = logging.getLogger(__name__)
 

bob/learn/tensorflow/loss/StyleLoss.py

@@ -5,7 +5,7 @@
 import logging
 import tensorflow as tf
 import functools
-logger = logging.getLogger("bob.learn.tensorflow")
+logger = logging.getLogger(__name__)
 
 
 def content_loss(noises, content_features):

bob/learn/tensorflow/loss/TripletLoss.py

@@ -3,7 +3,7 @@
 # @author: Tiago de Freitas Pereira <tiago.pereira@idiap.ch>
 
 import logging
-logger = logging.getLogger("bob.learn.tensorflow")
+logger = logging.getLogger(__name__)
 import tensorflow as tf
 
 from bob.learn.tensorflow.utils import compute_euclidean_distance

bob/learn/tensorflow/loss/mmd.py

@@ -2,6 +2,8 @@ import tensorflow as tf
 
 
 def compute_kernel(x, y):
+    """Gaussian kernel.
+    """
     x_size = tf.shape(x)[0]
     y_size = tf.shape(y)[0]
     dim = tf.shape(x)[1]
@@ -17,6 +19,9 @@ def compute_kernel(x, y):
 
 
 def mmd(x, y):
+    """Maximum Mean Discrepancy with Gaussian kernel.
+    See: https://stats.stackexchange.com/a/276618/49433
+    """
     x_kernel = compute_kernel(x, x)
     y_kernel = compute_kernel(y, y)
     xy_kernel = compute_kernel(x, y)