Fix skip connection in DGP example (#1531)

Co-authored-by: Geoff Pleiss <[email protected]>
cornellius-gp · Mar 29, 2021 · 2e55ec6 · 2e55ec6
1 parent 011679a
commit 2e55ec6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/05_Deep_Gaussian_Processes/Deep_Gaussian_Processes.ipynb b/examples/05_Deep_Gaussian_Processes/Deep_Gaussian_Processes.ipynb
@@ -192,7 +192,7 @@
     "                x = x.rsample()\n",
     "\n",
     "            processed_inputs = [\n",
-    "                inp.unsqueeze(0).expand(self.num_samples, *inp.shape)\n",
+    "                inp.unsqueeze(0).expand(gpytorch.settings.num_likelihood_samples.value(), *inp.shape)\n",
     "                for inp in other_inputs\n",
     "            ]\n",
     "\n",