5b3e337c7a67ea470dec1a4987eefb12168783f6,gpytorch/kernels/grid_interpolation_kernel.py,GridInterpolationKernel,__init__,#GridInterpolationKernel#Any#Any#Any#Any#,12

Before Change

        self.grid_bounds = grid_bounds

        if grid is None:
            grid = torch.zeros(len(grid_bounds), grid_size)
            for i in range(len(grid_bounds)):
                grid_diff = float(grid_bounds[i][1] - grid_bounds[i][0]) / (grid_size - 2)
                grid[i] = torch.linspace(grid_bounds[i][0] - grid_diff,
                                         grid_bounds[i][1] + grid_diff,

After Change

        self.base_kernel_module = base_kernel_module
        self.grid_size = grid_size
        self.register_buffer("grid", grid)
        self.register_buffer("_inducing_points", inducing_points)

    def train(self, mode=True):
        if hasattr(self, "_cached_kernel_mat"):
            del self._cached_kernel_mat

Instances