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

Before Change

            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))
            )

After Change

            batch_size (int): Number of batches.

        
        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))
            )

Instances