Update two_stream_shunt_adapter.py

two_stream_shunt_adapter.py

@@ -6,7 +6,7 @@ import torch.nn.functional as F
 
 # ─── Residual pocket block ────────────────────────────────────────────────
 class PocketBlock(nn.Module):
-    def __init__(self, dim, kernel=3, dropout=0.1):
+    def __init__(self, dim, kernel=3, dropout=0.0):
         super().__init__()
         self.body = nn.Sequential(
             nn.LayerNorm(dim),
@@ -34,7 +34,7 @@ class TwoStreamShuntAdapter(nn.Module):
         heads          = cfg["heads"]
         proj_layers    = cfg.get("proj_layers", 2)
         use_norm       = cfg.get("layer_norm", True)
-        p_drop         = cfg.get("dropout", 0.1)
+        p_drop         = cfg.get("dropout", 0.0)
         pocket_depth   = cfg.get("pocket_depth", 2)
 
         # helper ----------------------------------------------------------------