Minor aesthetic change

2021-05-17 16:57:41 +02:00 · 2021-05-17 16:57:41 +02:00 · e73b70ceb7
commit e73b70ceb7
parent 639198e774
1 changed files with 2 additions and 2 deletions
--- a/prototorch/functions/losses.py
+++ b/prototorch/functions/losses.py
@ -47,10 +47,10 @@ def lvq1_loss(distances, target_labels, prototype_labels):

 def lvq21_loss(distances, target_labels, prototype_labels):
    """LVQ2.1 loss function with support for one-hot labels.
-    
+
    See Section 4 [Sado&Yamada]
    https://papers.nips.cc/paper/1995/file/9c3b1830513cc3b8fc4b76635d32e692-Paper.pdf
    """
    dp, dm = _get_dp_dm(distances, target_labels, prototype_labels)
    mu = dp - dm
-    return mu
+    return mu