Update sequencer2D

liuruiyang98 · May 6, 2022 · b86656b · b86656b
1 parent 8feb07f
commit b86656b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/models_jittor/sequencer.py b/models_jittor/sequencer.py
@@ -72,7 +72,7 @@ def execute(self, x):
         return x
 
 class Sequencer2D(nn.Module):
-    def __init__(self, model_name: str = 'T', pretrained: str = None, num_classes: int = 1000, in_channels = 3, *args, **kwargs) -> None:
+    def __init__(self, model_name: str = 'M', pretrained: str = None, num_classes: int = 1000, in_channels = 3, *args, **kwargs) -> None:
         super().__init__()
         assert model_name in sequencer_settings.keys(), f"Sequencer model name should be in {list(sequencer_settings.keys())}"
         depth, embed_dims, hidden_dims, expansion_factor = sequencer_settings[model_name]

diff --git a/models_pytorch/sequencer.py b/models_pytorch/sequencer.py
@@ -72,7 +72,7 @@ def forward(self, x):
         return x
 
 class Sequencer2D(nn.Module):
-    def __init__(self, model_name: str = 'T', pretrained: str = None, num_classes: int = 1000, in_channels = 3, *args, **kwargs) -> None:
+    def __init__(self, model_name: str = 'M', pretrained: str = None, num_classes: int = 1000, in_channels = 3, *args, **kwargs) -> None:
         super().__init__()
         assert model_name in sequencer_settings.keys(), f"Sequencer model name should be in {list(sequencer_settings.keys())}"
         depth, embed_dims, hidden_dims, expansion_factor = sequencer_settings[model_name]