def __init__(self, dim, projection_size, hidden_size=4096):
    super().__init__()
    self.net = nn.Sequential(nn.Linear(dim, hidden_size), nn.BatchNorm1d(hidden_size), nn.ReLU(inplace=True), nn.Linear(hidden_size, projection_size))
def __init__(self, dim, projection_size, hidden_size=4096):
    super().__init__()
    self.net = nn.Sequential(nn.Linear(dim, hidden_size), nn.BatchNorm1d(hidden_size), nn.ReLU(inplace=True), nn.Linear(hidden_size, projection_size))