199db3617508c0512e3626ff20a0fcd257dae8d2,gpytorch/likelihoods/multitask_gaussian_likelihood.py,MultitaskGaussianLikelihood,__init__,#MultitaskGaussianLikelihood#Any#Any#Any#Any#Any#,27

Before Change

        super(MultitaskGaussianLikelihood, self).__init__(batch_size=batch_size, log_noise_prior=log_noise_prior)

        if rank == 0:
            self.register_parameter(
                name="log_task_noises",
                parameter=torch.nn.Parameter(torch.zeros(batch_size, num_tasks)),
                prior=task_prior,
            )
        else:
            self.register_parameter(
                name="task_noise_covar_factor", parameter=torch.nn.Parameter(torch.randn(batch_size, num_tasks, rank))
            )

After Change

                    parameter_names=("task_noise_corr_factor",),
                    transform=_eval_corr_matrix,
                )
        elif task_correlation_prior is not None:
            raise ValueError("Can only specify task_correlation_prior if rank>0")
        self.num_tasks = num_tasks

    def forward(self, input, *params):
        

Instances