add_back_item()

Nixtla · Sep 27, 2024 · e08ed44 · e08ed44
1 parent 536e8f6
commit e08ed44
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 18 deletions.
diff --git a/nbs/common.base_multivariate.ipynb b/nbs/common.base_multivariate.ipynb
@@ -390,12 +390,12 @@
     "\n",
     "        self.log(\n",
     "            'train_loss',\n",
-    "            loss.detach(),\n",
+    "            loss.detach().item(),\n",
     "            batch_size=outsample_y.size(0),\n",
     "            prog_bar=True,\n",
     "            on_epoch=True,\n",
     "        )\n",
-    "        self.train_trajectories.append((self.global_step, loss.detach()))\n",
+    "        self.train_trajectories.append((self.global_step, loss.detach().item()))\n",
     "        return loss\n",
     "\n",
     "    def validation_step(self, batch, batch_idx):\n",
@@ -440,7 +440,7 @@
     "\n",
     "        self.log(\n",
     "            'valid_loss',\n",
-    "            valid_loss.detach(),\n",
+    "            valid_loss.detach().item(),\n",
     "            batch_size=outsample_y.size(0),\n",
     "            prog_bar=True,\n",
     "            on_epoch=True,\n",

diff --git a/nbs/common.base_recurrent.ipynb b/nbs/common.base_recurrent.ipynb
@@ -365,12 +365,12 @@
     "\n",
     "        self.log(\n",
     "            'train_loss',\n",
-    "            loss.detach(),\n",
+    "            loss.detach().item(),\n",
     "            batch_size=outsample_y.size(0),\n",
     "            prog_bar=True,\n",
     "            on_epoch=True,\n",
     "        )\n",
-    "        self.train_trajectories.append((self.global_step, loss.detach()))\n",
+    "        self.train_trajectories.append((self.global_step, loss.detach().item()))\n",
     "        return loss\n",
     "\n",
     "    def validation_step(self, batch, batch_idx):\n",
@@ -438,7 +438,7 @@
     "\n",
     "        self.log(\n",
     "            'valid_loss',\n",
-    "            valid_loss.detach(),\n",
+    "            valid_loss.detach().item(),\n",
     "            batch_size=outsample_y.size(0),\n",
     "            prog_bar=True,\n",
     "            on_epoch=True,\n",

diff --git a/nbs/common.base_windows.ipynb b/nbs/common.base_windows.ipynb
@@ -447,12 +447,12 @@
     "\n",
     "        self.log(\n",
     "            'train_loss',\n",
-    "            loss.detach(),\n",
+    "            loss.detach().item(),\n",
     "            batch_size=outsample_y.size(0),\n",
     "            prog_bar=True,\n",
     "            on_epoch=True,\n",
     "        )\n",
-    "        self.train_trajectories.append((self.global_step, loss.detach()))\n",
+    "        self.train_trajectories.append((self.global_step, loss.detach().item()))\n",
     "        return loss\n",
     "\n",
     "    def _compute_valid_loss(self, outsample_y, output, outsample_mask, temporal_cols, y_idx):\n",
@@ -533,7 +533,7 @@
     "\n",
     "        self.log(\n",
     "            'valid_loss',\n",
-    "            valid_loss.detach(),\n",
+    "            valid_loss.detach().item(),\n",
     "            batch_size=batch_size,\n",
     "            prog_bar=True,\n",
     "            on_epoch=True,\n",

diff --git a/neuralforecast/common/_base_multivariate.py b/neuralforecast/common/_base_multivariate.py
@@ -389,12 +389,12 @@ def training_step(self, batch, batch_idx):
 
         self.log(
             "train_loss",
-            loss.detach(),
+            loss.detach().item(),
             batch_size=outsample_y.size(0),
             prog_bar=True,
             on_epoch=True,
         )
-        self.train_trajectories.append((self.global_step, loss.detach()))
+        self.train_trajectories.append((self.global_step, loss.detach().item()))
         return loss
 
     def validation_step(self, batch, batch_idx):
@@ -456,7 +456,7 @@ def validation_step(self, batch, batch_idx):
 
         self.log(
             "valid_loss",
-            valid_loss.detach(),
+            valid_loss.detach().item(),
             batch_size=outsample_y.size(0),
             prog_bar=True,
             on_epoch=True,

diff --git a/neuralforecast/common/_base_recurrent.py b/neuralforecast/common/_base_recurrent.py
@@ -349,12 +349,12 @@ def training_step(self, batch, batch_idx):
 
         self.log(
             "train_loss",
-            loss.detach(),
+            loss.detach().item(),
             batch_size=outsample_y.size(0),
             prog_bar=True,
             on_epoch=True,
         )
-        self.train_trajectories.append((self.global_step, loss.detach()))
+        self.train_trajectories.append((self.global_step, loss.detach().item()))
         return loss
 
     def validation_step(self, batch, batch_idx):
@@ -447,7 +447,7 @@ def validation_step(self, batch, batch_idx):
 
         self.log(
             "valid_loss",
-            valid_loss.detach(),
+            valid_loss.detach().item(),
             batch_size=outsample_y.size(0),
             prog_bar=True,
             on_epoch=True,

diff --git a/neuralforecast/common/_base_windows.py b/neuralforecast/common/_base_windows.py
@@ -440,12 +440,12 @@ def training_step(self, batch, batch_idx):
 
         self.log(
             "train_loss",
-            loss.detach(),
+            loss.detach().item(),
             batch_size=outsample_y.size(0),
             prog_bar=True,
             on_epoch=True,
         )
-        self.train_trajectories.append((self.global_step, loss.detach()))
+        self.train_trajectories.append((self.global_step, loss.detach().item()))
         return loss
 
     def _compute_valid_loss(
@@ -551,7 +551,7 @@ def validation_step(self, batch, batch_idx):
 
         self.log(
             "valid_loss",
-            valid_loss.detach(),
+            valid_loss.detach().item(),
             batch_size=batch_size,
             prog_bar=True,
             on_epoch=True,