fix: missing metrics for translations.

ruflab · Sep 7, 2023 · 4f3e2a0 · 4f3e2a0
1 parent 68ae8a3
commit 4f3e2a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "shimmer"
-version = "0.1.0"
+version = "0.1.1"
 description = "A light GLoW"
 authors = ["bdvllrs <[email protected]>"]
 license = "MIT"

diff --git a/shimmer/modules/losses.py b/shimmer/modules/losses.py
@@ -134,6 +134,7 @@ def _translation_loss(
                     }
                 )
     losses["translations"] = torch.stack(list(losses.values()), dim=0).mean()
+    losses.update(metrics)
     return losses
-Original file line number
+Diff line change
@@ Expand Up / @@ -134,6 +134,7 @@ def _translation_loss( @@
                         }
                     )
         losses["translations"] = torch.stack(list(losses.values()), dim=0).mean()
+        losses.update(metrics)
         return losses
@@ Expand Down @@