libs/solaris/nets/zoo/selim_sef_sn4.py [473:493]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def get_encoder(self, encoder, layer):
        if layer == 0:
            return nn.Sequential(
                encoder.features.conv0,  # conv
                encoder.features.norm0,  # bn
                encoder.features.relu0  # relu
            )
        elif layer == 1:
            return nn.Sequential(encoder.features.pool0,
                                 encoder.features.denseblock1)
        elif layer == 2:
            return nn.Sequential(encoder.features.transition1,
                                 encoder.features.denseblock2)
        elif layer == 3:
            return nn.Sequential(encoder.features.transition2,
                                 encoder.features.denseblock3)
        elif layer == 4:
            return nn.Sequential(encoder.features.transition3,
                                 encoder.features.denseblock4,
                                 encoder.features.norm5,
                                 nn.ReLU())
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



libs/solaris/nets/zoo/selim_sef_sn4.py [501:521]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def get_encoder(self, encoder, layer):
        if layer == 0:
            return nn.Sequential(
                encoder.features.conv0,  # conv
                encoder.features.norm0,  # bn
                encoder.features.relu0  # relu
            )
        elif layer == 1:
            return nn.Sequential(encoder.features.pool0,
                                 encoder.features.denseblock1)
        elif layer == 2:
            return nn.Sequential(encoder.features.transition1,
                                 encoder.features.denseblock2)
        elif layer == 3:
            return nn.Sequential(encoder.features.transition2,
                                 encoder.features.denseblock3)
        elif layer == 4:
            return nn.Sequential(encoder.features.transition3,
                                 encoder.features.denseblock4,
                                 encoder.features.norm5,
                                 nn.ReLU())
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -