Merge pull request #1647 from cornellius-gp/large_loss_multitask_fix

Fix erroneous loss for ExactGP multitask models
cornellius-gp · Jun 10, 2021 · a0d8cd2 · a0d8cd2
2 parents 7361840 + b5b20a2
commit a0d8cd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpytorch/mlls/exact_marginal_log_likelihood.py b/gpytorch/mlls/exact_marginal_log_likelihood.py
@@ -63,7 +63,7 @@ def forward(self, function_dist, target, *params):
         res = self._add_other_terms(res, params)
 
         # Scale by the amount of data we have
-        num_data = target.size(-1)
+        num_data = function_dist.event_shape.numel()
         return res.div_(num_data)
 
     def pyro_factor(self, output, target, *params):