Merge pull request #652 from sdaulton/fixed_noise_attr

Make noise an attribute in FixedGaussianNoise

Author: jacobrgardner

gpytorch/likelihoods/noise_models.py

@@ -153,7 +153,7 @@ def forward(
 class FixedGaussianNoise(Module):
     def __init__(self, noise: Tensor) -> None:
         super().__init__()
-        self.register_buffer("noise", noise)
+        self.noise = noise
 
     def forward(
         self,
@@ -172,3 +172,7 @@ def forward(
             return DiagLazyTensor(self.noise)
         else:
             return ZeroLazyTensor()
+
+    def _apply(self, fn):
+        self.noise = fn(self.noise)
+        return super(FixedGaussianNoise, self)._apply(fn)

test/examples/test_kissgp_white_noise_regression.py

@@ -131,7 +131,7 @@ def test_kissgp_gp_fast_pred_var(self):
 
             # Now bump up the likelihood to something huge
             # This will make it easy to calculate the variance
-            likelihood.initialize(noise=3.)
+            likelihood.noise = torch.ones(100) * 3.0
             test_function_predictions = likelihood(gp_model(train_x))
 
             noise = likelihood.noise