From ea15e6940e520eeb3d38c1c805e0a00b0df60dab Mon Sep 17 00:00:00 2001
From: Paul-Edouard Sarlin <15985472+sarlinpe@users.noreply.github.com>
Date: Wed, 24 Jan 2024 15:47:12 +0100
Subject: [PATCH] Fix typo (#46)

---
 gluefactory/models/utils/losses.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gluefactory/models/utils/losses.py b/gluefactory/models/utils/losses.py
index cca17636..06c7958b 100644
--- 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