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))
