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 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)