fix nearest_neighbor_variational_strategy type error

LuhuanWu · LuhuanWu · commit dc4657b2e6ac · 2023-07-08T23:34:29.000-07:00
diff --git a/gpytorch/variational/nearest_neighbor_variational_strategy.py b/gpytorch/variational/nearest_neighbor_variational_strategy.py
@@ -154,7 +154,7 @@ def __call__(self, x: Tensor, prior: bool = False, **kwargs: Any) -> Multivariat
                     torch.randn_like(prior_dist.mean), alpha=self._variational_distribution.mean_init_std
                 )
                 # initialize with a small variational stddev for quicker conv. of kl divergence
-                self._variational_distribution._variational_stddev.data.copy_(1e-2)
+                self._variational_distribution._variational_stddev.data.copy_(torch.tensor(1e-2))
                 self.variational_params_initialized.fill_(1)
 
             return self.forward(

Original file line number	Diff line number	Diff line change
`@@ -154,7 +154,7 @@ def __call__(self, x: Tensor, prior: bool = False, **kwargs: Any) -> Multivariat`
`154`	`154`	`torch.randn_like(prior_dist.mean), alpha=self._variational_distribution.mean_init_std`
`155`	`155`	`)`
`156`	`156`	`# initialize with a small variational stddev for quicker conv. of kl divergence`
`157`		`- self._variational_distribution._variational_stddev.data.copy_(1e-2)`
	`157`	`+ self._variational_distribution._variational_stddev.data.copy_(torch.tensor(1e-2))`
`158`	`158`	`self.variational_params_initialized.fill_(1)`
`159`	`159`
`160`	`160`	`return self.forward(`