Lighten TimesNet exposure backbone

models.py

@@ -33,12 +33,12 @@ class DeepHealth(nn.Module):
         use_exposure_encoder: bool = False,
         exposure_daily_input_dim: int = 4,
         exposure_monthly_input_dim: int = 2,
-        exposure_d_model: int | None = None,
+        exposure_d_model: int | None = 64,
         exposure_n_layers: int = 2,
-        exposure_top_k: int = 3,
-        exposure_n_convnext_blocks: int = 2,
-        exposure_conv_kernel_size: int = 7,
-        exposure_mlp_ratio: float = 4.0,
+        exposure_top_k: int = 2,
+        exposure_n_backbone_blocks: int = 1,
+        exposure_backbone_kernel_size: int = 5,
+        exposure_backbone_expansion: float = 2.0,
         exposure_use_gate: bool = True,
     ):
         super().__init__()
@@ -64,9 +64,9 @@ class DeepHealth(nn.Module):
                 d_model=exposure_d_model,
                 n_layers=exposure_n_layers,
                 top_k=exposure_top_k,
-                n_convnext_blocks=exposure_n_convnext_blocks,
-                conv_kernel_size=exposure_conv_kernel_size,
-                mlp_ratio=exposure_mlp_ratio,
+                n_backbone_blocks=exposure_n_backbone_blocks,
+                backbone_kernel_size=exposure_backbone_kernel_size,
+                backbone_expansion=exposure_backbone_expansion,
                 dropout=dropout,
                 use_gate=exposure_use_gate,
             )