remove_unused_layer_nbeatsx

Nixtla · Sep 30, 2024 · 461e260 · 461e260
1 parent 69255e8
commit 461e260
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 12 deletions.
diff --git a/nbs/models.nbeatsx.ipynb b/nbs/models.nbeatsx.ipynb
@@ -524,12 +524,6 @@
     "        )\n",
     "        self.blocks = torch.nn.ModuleList(blocks)\n",
     "\n",
-    "        # Adapter with Loss dependent dimensions\n",
-    "        if self.loss.outputsize_multiplier > 1:\n",
-    "            self.out = nn.Linear(\n",
-    "                in_features=h, out_features=h * self.loss.outputsize_multiplier\n",
-    "            )\n",
-    "\n",
     "    def create_stack(\n",
     "        self,\n",
     "        h,\n",

diff --git a/neuralforecast/models/nbeatsx.py b/neuralforecast/models/nbeatsx.py
@@ -427,12 +427,6 @@ def __init__(
         )
         self.blocks = torch.nn.ModuleList(blocks)
 
-        # Adapter with Loss dependent dimensions
-        if self.loss.outputsize_multiplier > 1:
-            self.out = nn.Linear(
-                in_features=h, out_features=h * self.loss.outputsize_multiplier
-            )
-
     def create_stack(
         self,
         h,