Merge pull request #8 from chjort/transformer_dynamic_input

Transformer dynamic input
chjort · Apr 3, 2021 · 2c08c79 · 2c08c79
2 parents 864f4d4 + 151ce9a
commit 2c08c79
Show file tree

Hide file tree

Showing 2 changed files with 230 additions and 58 deletions.
diff --git a/chambers/models/backbones/__init__.py b/chambers/models/backbones/__init__.py
@@ -7,4 +7,11 @@
 from .senet import SEResNeXt50
 from .senet import SEResNeXt101
 from .senet import SENet154
+from .vision_transformer import ViTS16
+from .vision_transformer import ViTB16
+from .vision_transformer import ViTB32
+from .vision_transformer import ViTL16
+from .vision_transformer import ViTL32
+from .vision_transformer import DeiTS16
+from .vision_transformer import DeiTB16