diff --git a/znnl/loss_functions/wasserstein_loss.py b/znnl/loss_functions/wasserstein_loss.py index 3caf2d2..a04921c 100644 --- a/znnl/loss_functions/wasserstein_loss.py +++ b/znnl/loss_functions/wasserstein_loss.py @@ -49,7 +49,12 @@ def __init__(self): """ Constructor for the Wasserstein loss class. """ - logger.warning("The Wasserstein loss cannot be used for gradient calculations.") + logger.warning( + "The Wasserstein loss cannot be used for gradient calculations, " + "and therefore not for neural network training. " + "It can be used to compare two distributions, however, the current " + "implementation is not differentiable!" + ) super(WassersteinLoss, self).__init__() self.metric = WassersteinDistance()