Upload 2 files

spectral/notebooks/experiment_5_matrix_decompositions.ipynb

@@ -1970,6 +1970,7 @@
     "        self.output_dim = 3 * 3 + self.pool_size * self.pool_size  # 3\u00d73 channel gram + spatial eigen\n",
     "        print(f\"[EIGEN] dim={self.output_dim}\")\n",
     "\n",
+    "    @torch.amp.custom_fwd(device_type='cuda', cast_inputs=torch.float32)\n",
     "    def forward(self, x):\n",
     "        B, C, H, W = x.shape\n",
     "        # Channel gram matrix\n",

spectral/notebooks/experiment_7_composite_pipelines.ipynb

@@ -2036,6 +2036,7 @@
     "        self.output_dim = self.scatt_dim + self.stft_dim + self.eigen_dim\n",
     "        print(f\"[MULTISIG] scatt={self.scatt_dim} + stft={self.stft_dim} + eigen={self.eigen_dim} = {self.output_dim}\")\n",
     "\n",
+    "    @torch.amp.custom_fwd(device_type='cuda', cast_inputs=torch.float32)\n",
     "    def forward(self, x):\n",
     "        B, C, H, W = x.shape\n",
     "        # Scattering\n",