529456c17464022d42e1c343bec52f2d4d77e79b,gpytorch/likelihoods/multitask_gaussian_likelihood.py,HomoskedasticMultitaskGaussianLikelihood,__init__,#HomoskedasticMultitaskGaussianLikelihood#Any#Any#Any#Any#Any#,135

Before Change

        self.register_parameter(
            name="log_noise", parameter=torch.nn.Parameter(torch.zeros(batch_size, 1)), 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))
            )
            if task_prior is not None:
                self.register_derived_prior(
                    name="MultitaskErrorCovariancePrior",
                    prior=task_prior,
                    parameter_names=("task_noise_covar_factor", "log_noise"),
                    transform=_eval_covar_matrix,
                )
        self.num_tasks = num_tasks

    def forward(self, input):
        

After Change

        
        super(_MultitaskGaussianLikelihoodBase, self).__init__(log_noise_covar=log_noise_covar)
        if rank != 0:
            self.register_parameter(
                name="task_noise_corr_factor", parameter=torch.nn.Parameter(torch.randn(batch_size, num_tasks, rank))
            )
            if task_correlation_prior is not None:
                self.register_derived_prior(
                    name="MultitaskErrorCorrelationPrior",
                    prior=task_correlation_prior,
                    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