Transformer Implementation

2020-10-29 16:40:43 +01:00
parent b57eabb371
commit 7bac9e984b
4 changed files with 103 additions and 2 deletions
--- a/util/config.py
+++ b/util/config.py
@@ -4,6 +4,7 @@ from models.bandwise_conv_classifier import BandwiseConvClassifier
 from models.bandwise_conv_multihead_classifier import BandwiseConvMultiheadClassifier
 from models.ensemble import Ensemble
 from models.residual_conv_classifier import ResidualConvClassifier
+from models.transformer_model import VisualTransformer


 class MConfig(Config):
@@ -20,5 +21,6 @@ class MConfig(Config):
                    Ensemble=Ensemble,
                    E=Ensemble,
                    ResidualConvClassifier=ResidualConvClassifier,
-                    RCC=ResidualConvClassifier
+                    RCC=ResidualConvClassifier,
+                    ViT=VisualTransformer
                    )