Feature/sg 109 beit architecture (#185)

* beit added * url updated and redundant files removed * beit large conf fixed * linter * replace head support * todo comment for method removal * redundant method removed
Deci-AI · Jun 8, 2022 · bd56ade · bd56ade
1 parent 658f638
commit bd56ade
Show file tree

Hide file tree

Showing 4 changed files with 474 additions and 4 deletions.
diff --git a/src/super_gradients/training/models/all_architectures.py b/src/super_gradients/training/models/all_architectures.py
@@ -22,7 +22,7 @@
     STDC1Seg, STDC2Seg
 
 from super_gradients.training.models.kd_modules.kd_module import KDModule
-
+from super_gradients.training.models.classification_models.beit import beit_base_patch16_224, beit_large_patch16_224
 # IF YOU UPDATE THE ARCHITECTURE DICT PLEASE ALSO UPDATE THE ENUM CLASS DOWN BELOW.
 
 
@@ -115,6 +115,8 @@
                  "vit_base": vit_base,
                  "vit_large": vit_large,
                  "vit_huge": vit_huge,
+                 "beit_base_patch16_224": beit_base_patch16_224,
+                 "beit_large_patch16_224": beit_large_patch16_224
                  }
 
 KD_ARCHITECTURES = {