def get_encoder(self, encoder, layer): features = encoder.features if layer == 0: return nn.Sequential(*get_slice(features, 0, 2)) if layer == 1: return nn.Sequential(*get_slice(features, 2, 5)) if layer == 2: return nn.Sequential(*get_slice(features, 5, 10)) if layer == 3: return nn.Sequential(*get_slice(features, 10, 15)) if layer == 4: return nn.Sequential(*get_slice(features, 15, 20))
def get_encoder(self, encoder, layer): features = encoder.features if layer == 0: return nn.Sequential(*get_slice(features, 0, 3)) if layer == 1: return nn.Sequential(*get_slice(features, 3, 7)) if layer == 2: return nn.Sequential(*get_slice(features, 7, 14)) if layer == 3: return nn.Sequential(*get_slice(features, 14, 20)) if layer == 4: return nn.Sequential(*get_slice(features, 20, 27))
def get_encoder(self, encoder, layer): features = encoder.features if layer == 0: return nn.Sequential(*get_slice(features, 0, 6)) if layer == 1: return nn.Sequential(*get_slice(features, 6, 13)) if layer == 2: return nn.Sequential(*get_slice(features, 13, 23)) if layer == 3: return nn.Sequential(*get_slice(features, 23, 33)) if layer == 4: return nn.Sequential(*get_slice(features, 33, 43))
def get_encoder(self, encoder, layer): features = encoder.features if layer == 0: return nn.Sequential(*get_slice(features, 0, 4)) if layer == 1: return nn.Sequential(*get_slice(features, 4, 9)) if layer == 2: return nn.Sequential(*get_slice(features, 9, 14)) if layer == 3: return nn.Sequential(*get_slice(features, 14, 19)) if layer == 4: return nn.Sequential(*get_slice(features, 19, 24))