def __init__(self, in_channels, representation_size):
    super().__init__()
    self.fc6 = nn.Linear(in_channels, representation_size)
    self.fc7 = nn.Linear(representation_size, representation_size)
def __init__(self, in_channels, representation_size):
    super().__init__()
    self.fc6 = nn.Linear(in_channels, representation_size)
    self.fc7 = nn.Linear(representation_size, representation_size)