fix small bug in config

OpenGVLab · Mar 30, 2024 · f3427e4 · f3427e4
1 parent 89e2de9
commit f3427e4
Show file tree

Hide file tree

Showing 11 changed files with 12 additions and 12 deletions.
diff --git a/videomamba/video_mm/exp_pt/videomamba_middle_17m/config.py b/videomamba/video_mm/exp_pt/videomamba_middle_17m/config.py
@@ -47,7 +47,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_pt/videomamba_middle_17m_unmasked/config.py b/videomamba/video_mm/exp_pt/videomamba_middle_17m_unmasked/config.py
@@ -47,7 +47,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_pt/videomamba_middle_25m/config.py b/videomamba/video_mm/exp_pt/videomamba_middle_25m/config.py
@@ -47,7 +47,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_pt/videomamba_middle_25m_unmasked/config.py b/videomamba/video_mm/exp_pt/videomamba_middle_25m_unmasked/config.py
@@ -47,7 +47,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_pt/videomamba_middle_5m/config.py b/videomamba/video_mm/exp_pt/videomamba_middle_5m/config.py
@@ -47,7 +47,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_zs/anet/config.py b/videomamba/video_mm/exp_zs/anet/config.py
@@ -59,7 +59,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_zs/didemo/config.py b/videomamba/video_mm/exp_zs/didemo/config.py
@@ -65,7 +65,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_zs/lsmdc/config.py b/videomamba/video_mm/exp_zs/lsmdc/config.py
@@ -59,7 +59,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_zs/msrvtt/config.py b/videomamba/video_mm/exp_zs/msrvtt/config.py
@@ -59,7 +59,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/exp_zs/msvd/config.py b/videomamba/video_mm/exp_zs/msvd/config.py
@@ -65,7 +65,7 @@
         fused_add_norm=True,
         rms_norm=True, 
         residual_in_fp32=True,
-        bimamba_type="v2",
+        bimamba=True,
         pool_type="cls+avg",
         kernel_size=1,
         num_frames="${num_frames}",

diff --git a/videomamba/video_mm/models/backbones/videomamba/videomamba.py b/videomamba/video_mm/models/backbones/videomamba/videomamba.py
@@ -544,7 +544,7 @@ def build_videomamba(config, add_pool_norm=True):
         fused_add_norm=config.vision_encoder.fused_add_norm,
         rms_norm=config.vision_encoder.rms_norm,
         residual_in_fp32=config.vision_encoder.residual_in_fp32,
-        bimamba_type=config.vision_encoder.bimamba_type,
+        bimamba=config.vision_encoder.bimamba,
         pool_type=config.vision_encoder.pool_type,
         kernel_size=config.vision_encoder.kernel_size,
         num_frames=config.vision_encoder.num_frames,
@@ -595,7 +595,7 @@ def build_videomamba(config, add_pool_norm=True):
             "fused_add_norm": True,
             "rms_norm": True, 
             "residual_in_fp32": True,
-            "bimamba_type": "v2",
+            "bimamba": True,
             "pool_type": "cls",
             "kernel_size": 1,
             "num_frames": 8,