diff --git a/bob/learn/tensorflow/loss/BaseLoss.py b/bob/learn/tensorflow/loss/BaseLoss.py
index 5b30e18e47a9eeba280ab2110e89e940eca69cf5..50e83e14a089466c25933301d56504a0ad66ce9b 100644
--- a/bob/learn/tensorflow/loss/BaseLoss.py
+++ b/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
 
diff --git a/bob/learn/tensorflow/loss/ContrastiveLoss.py b/bob/learn/tensorflow/loss/ContrastiveLoss.py
index 93612cbfd394ce6e87454ef6e1e4179c50a3f942..6fa29f1aefe140d6e09de5c876958735f5e5508b 100644
--- a/bob/learn/tensorflow/loss/ContrastiveLoss.py
+++ b/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__)
 
diff --git a/bob/learn/tensorflow/loss/StyleLoss.py b/bob/learn/tensorflow/loss/StyleLoss.py
index 36b53b66dcd70d94693c856129f7b23a303d68ab..43cc655e291d5634f4cbe2032f6ebcf154b2a465 100644
--- a/bob/learn/tensorflow/loss/StyleLoss.py
+++ b/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):
diff --git a/bob/learn/tensorflow/loss/TripletLoss.py b/bob/learn/tensorflow/loss/TripletLoss.py
index 2d487cc5f89f41b44f36767a6cc05fc8a5b3d5e1..d2616d6aa0394f1ea024b65b9199292221951ca4 100644
--- a/bob/learn/tensorflow/loss/TripletLoss.py
+++ b/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
diff --git a/bob/learn/tensorflow/loss/mmd.py b/bob/learn/tensorflow/loss/mmd.py
index bd7df3e52b248aeb992c318148a8ec16f6a7476d..2933d7b1d3cd32b7533c2fa5213e38eeb7192965 100644
--- a/bob/learn/tensorflow/loss/mmd.py
+++ b/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)