Replace a use of _dccn with .item()

Bitbol-Lab · Dec 15, 2023 · 655ad95 · 655ad95
1 parent e061313
commit 655ad95
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/diffpass/train.py b/diffpass/train.py
@@ -594,9 +594,9 @@ def _prepare_fit(self, x: torch.Tensor, y: torch.Tensor) -> DiffPASSResults:
             results.soft_losses_identity_perm[
                 self.information_measure
             ] = results.hard_losses_identity_perm[self.information_measure]
-            results.soft_losses_identity_perm["BestHits"] = _dccn(
-                self.inter_group_loss(self._bh_soft_x, self._bh_soft_y)
-            )
+            results.soft_losses_identity_perm["BestHits"] = self.inter_group_loss(
+                self._bh_soft_x, self._bh_soft_y
+            ).item()
 
         return results
 

diff --git a/nbs/train.ipynb b/nbs/train.ipynb
@@ -674,9 +674,9 @@
     "            results.soft_losses_identity_perm[\n",
     "                self.information_measure\n",
     "            ] = results.hard_losses_identity_perm[self.information_measure]\n",
-    "            results.soft_losses_identity_perm[\"BestHits\"] = _dccn(\n",
-    "                self.inter_group_loss(self._bh_soft_x, self._bh_soft_y)\n",
-    "            )\n",
+    "            results.soft_losses_identity_perm[\"BestHits\"] = self.inter_group_loss(\n",
+    "                self._bh_soft_x, self._bh_soft_y\n",
+    "            ).item()\n",
     "\n",
     "        return results\n",
     "\n",