Detach indices in Interpolation()

cornellius-gp · Apr 16, 2018 · 8b50f2f · 8b50f2f
1 parent f70f52f
commit 8b50f2f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gpytorch/utils/interpolation.py b/gpytorch/utils/interpolation.py
@@ -78,6 +78,7 @@ def interpolate(self, x_grid, x_target, interp_points=range(-2, 2)):
             lower_grid_pt_idxs = torch.floor((x_target[:, i] - x_grid[i, 0]) / grid_delta).squeeze()
             lower_pt_rel_dists = (x_target[:, i] - x_grid[i, 0]) / grid_delta - lower_grid_pt_idxs
             lower_grid_pt_idxs = lower_grid_pt_idxs - interp_points.max()
+            lower_grid_pt_idxs.detach_()
 
             scaled_dist = lower_pt_rel_dists.unsqueeze(-1) + interp_points_flip.unsqueeze(-2)
             dim_interp_values = self._cubic_interpolation_kernel(scaled_dist)