ResidualModule and New Parameters, Speed Manipulation

2020-05-12 12:37:26 +02:00
parent 3fbc98dfa3
commit 28bfcfdce3
8 changed files with 181 additions and 78 deletions
--- a/util/config.py
+++ b/util/config.py
@ -3,6 +3,7 @@ from models.conv_classifier import ConvClassifier
 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


 class MConfig(Config):
@ -11,7 +12,13 @@ class MConfig(Config):
    @property
    def _model_map(self):
        return dict(ConvClassifier=ConvClassifier,
+                    CC=ConvClassifier,
                    BandwiseConvClassifier=BandwiseConvClassifier,
+                    BCC=BandwiseConvClassifier,
                    BandwiseConvMultiheadClassifier=BandwiseConvMultiheadClassifier,
+                    BCMC=BandwiseConvMultiheadClassifier,
                    Ensemble=Ensemble,
+                    E=Ensemble,
+                    ResidualConvClassifier=ResidualConvClassifier,
+                    RCC=ResidualConvClassifier
                    )