Fix typo (#46)

cvg · Jan 24, 2024 · 365ad7b · 365ad7b
1 parent bfdae2c
commit 365ad7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gluefactory/models/utils/losses.py b/gluefactory/models/utils/losses.py
@@ -69,5 +69,5 @@ def nll_loss(self, log_assignment, data):
         weights[:, :m, :n] = positive
 
         weights[:, :m, -1] = neg0
-        weights[:, -1, :m] = neg1
+        weights[:, -1, :n] = neg1
         return weights