def __init__(self, vgg_model):
    super(LossNetwork, self).__init__()
    self.vgg_layers = vgg_model
    self.layer_name_mapping = {'3': 'relu1_2', '8': 'relu2_2', '15': 'relu3_3'}
def __init__(self, vgg_model):
    super(LossNetwork, self).__init__()
    self.vgg_layers = vgg_model
    self.layer_name_mapping = {'3': 'relu1_2', '8': 'relu2_2', '15': 'relu3_3'}