We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8125eaa commit ba238fbCopy full SHA for ba238fb
test/kernels/test_inducing_point_kernel.py
@@ -50,10 +50,12 @@ def test_kernel_output(self):
50
self.assertTrue(ps_mock.called)
51
52
# Check whether changing diagonal correction makes a difference (ensuring that cache is cleared)
53
- model.train(); model.eval();
+ model.train()
54
+ model.eval()
55
with gpytorch.settings.sgpr_diagonal_correction(True), torch.no_grad():
56
output_mean_correct = model(test_x).mean
57
58
59
with gpytorch.settings.sgpr_diagonal_correction(False), torch.no_grad():
60
output_mean_no_correct = model(test_x).mean
61
self.assertNotAlmostEqual(output_mean_correct.sum().item(), output_mean_no_correct.sum().item())
0 commit comments