diff --git a/docs/tutorials/fit_textured_mesh.ipynb b/docs/tutorials/fit_textured_mesh.ipynb index 83ad5c73..4e3287ad 100644 --- a/docs/tutorials/fit_textured_mesh.ipynb +++ b/docs/tutorials/fit_textured_mesh.ipynb @@ -455,7 +455,8 @@ " target_image=target_rgb[1], title='', \n", " silhouette=False):\n", " inds = 3 if silhouette else range(3)\n", - " predicted_images = renderer(predicted_mesh)\n", + " with torch.no_grad():\n", + " predicted_images = renderer(predicted_mesh)\n", " plt.figure(figsize=(20, 10))\n", " plt.subplot(1, 2, 1)\n", " plt.imshow(predicted_images[0, ..., inds].cpu().detach().numpy())\n",