(self, in_channels, mlp_hidden_size, projection_size): super(MLPHead, self).__init__() self.net = nn.Sequential( nn.Linear(in_channels, mlp_hidden_size), nn.BatchNorm1d(mlp_hidden_size), nn.ReLU(inplace=True), nn.Linear(mlp_hidden_size, projection_size) nn.BatchNorm1d(mlp_...