Update model.py

model.py

@@ -4,7 +4,7 @@ import torch.nn.functional as F
 
 # ─── Residual Pocket Block ───────────────────────────────────
 class BottleneckResBlock(nn.Module):
-    def __init__(self, dim, kernel=3, dropout=0.1):
+    def __init__(self, dim, kernel=3, dropout=0.0):
         super().__init__()
         self.norm = nn.LayerNorm(dim)
         self.conv = nn.Conv1d(dim, dim, kernel_size=kernel, padding=kernel // 2, groups=1)
@@ -36,7 +36,7 @@ class TwoStreamShuntAdapter(nn.Module):
 
         use_norm   = config.get("layer_norm", True)
         use_do     = config.get("use_dropout", True)
-        do_p       = config.get("dropout", 0.1)
+        do_p       = config.get("dropout", 0.0)
         proj_depth = config.get("proj_layers", 2)
 
         def build_projection(input_dim, output_dim):