Upload xformer.py

rscd/models/decoderheads/xformer.py

@@ -400,9 +400,9 @@ class CTTF1(nn.Module):
         self.convB    = nn.Conv2d(in_channel, 1, 1)
         self.sigmoid  = nn.Sigmoid()
 
-        self.mixer = Residual(WAVE_mamba(in_channel, global_ratio, local_ratio,pa_ratio, kernels, ssm_ratio, forward_type))
+        self.mixer = Residual(GLP_block(in_channel, global_ratio, local_ratio,pa_ratio, kernels, ssm_ratio, forward_type))
         self.mixer2 = Residual(
-            SAS(in_channel, global_ratio = 0, local_ratio = 0.1, pa_ratio = 0, kernels = 5, ssm_ratio = 1, forward_type = "v052d"))
+            SASF(in_channel, global_ratio = 0, local_ratio = 0.1, pa_ratio = 0, kernels = 5, ssm_ratio = 1, forward_type = "v052d"))
 
         self.fuse = nn.Sequential(
         nn.Conv2d(in_channel * 3, in_channel, kernel_size=1),
@@ -445,9 +445,9 @@ class CTTF2(nn.Module):
 
 
         self.mixer = Residual(
-            WAVE_mamba(in_channel, global_ratio, local_ratio, pa_ratio, kernels, ssm_ratio, forward_type))
+            GLP_block(in_channel, global_ratio, local_ratio, pa_ratio, kernels, ssm_ratio, forward_type))
         self.mixer2 = Residual(
-            SAS(in_channel, global_ratio=0, local_ratio=0.1, pa_ratio=0, kernels=5, ssm_ratio=1,
+            SASF(in_channel, global_ratio=0, local_ratio=0.1, pa_ratio=0, kernels=5, ssm_ratio=1,
                               forward_type="v052d"))
 
         self.fuse = nn.Sequential(