Fix (long deprecated) failing torch.linspace without steps arg example`
"# and then trains/tests an approximate GP with it on the supplied dataset\n",
"\n",
"def train_and_test_approximate_gp(objective_function_cls):\n",
- " model = ApproximateGPModel(torch.linspace(0, 1))\n",
+ " model = ApproximateGPModel(torch.linspace(0, 1, 100))\n",
" likelihood = gpytorch.likelihoods.GaussianLikelihood()\n",
" objective_function = objective_function_cls(likelihood, model, num_data=train_y.numel())\n",
" optimizer = torch.optim.Adam(list(model.parameters()) + list(likelihood.parameters()), lr=0.1)\n",
