def output_features(self, x):
    output = {}
    for (name, module) in self.vgg_layers._modules.items():
        x = module(x)
        if (name in self.layer_name_mapping):
            output[self.layer_name_mapping[name]] = x
    return list(output.values())
def output_features(self, x):
    output = {}
    for (name, module) in self.vgg_layers._modules.items():
        x = module(x)
        if (name in self.layer_name_mapping):
            output[self.layer_name_mapping[name]] = x
    return list(output.values())