rm unrecognized args from basic layer

innat · Mar 24, 2024 · c4e8201 · c4e8201
1 parent 4ba6d39
commit c4e8201
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/videoswin/blocks/basic.py b/videoswin/blocks/basic.py
@@ -148,7 +148,6 @@ def get_config(self):
                 "window_size": self.window_size,
                 "num_heads": self.num_heads,
                 "mlp_ratio": self.mlp_ratio,
-                "shift_size": self.shift_size,
                 "depth": self.depth,
                 "qkv_bias": self.qkv_bias,
                 "qk_scale": self.qk_scale,
@@ -158,3 +157,4 @@ def get_config(self):
             }
         )
         return config
+