Rename cobra.py to mcdpmamba.py

cobra.py → mcdpmamba.py

@@ -27,7 +27,7 @@ class FeedForward(nn.Module):
 
 
 
-class COBRAGatingUnit(nn.Module):
+class MCGatingUnit(nn.Module):
     def __init__(self,d_model,d_ffn,dropout):
         super().__init__()
         
@@ -47,12 +47,12 @@ class COBRAGatingUnit(nn.Module):
         return out
 
 
-class COBRABlock(nn.Module):
+class MCDPMAMBABlock(nn.Module):
     def __init__(self, d_model, d_ffn,dropout):
         super().__init__()
        
         self.norm = nn.LayerNorm(d_model)       
-        self.cobgu = COBRAGatingUnit(d_model,d_ffn,dropout)
+        self.mcgu = MCGatingUnit(d_model,d_ffn,dropout)
         self.ffn = FeedForward(d_model,d_ffn,dropout)
     def forward(self, x):
         residual = x
@@ -73,13 +73,13 @@ class COBRABlock(nn.Module):
 
 
 
-class COBRA(nn.Module):
+class MCDPMAMBA(nn.Module):
     def __init__(self, d_model, d_ffn, num_layers,dropout):
         super().__init__()
         
         self.model = nn.Sequential(
             
-            *[COBRABlock(d_model,d_ffn,dropout) for _ in range(num_layers)],
+            *[MCDPMAMBABlock(d_model,d_ffn,dropout) for _ in range(num_layers)],
             
             
         )