diff --git a/docs/tutorials/05_torch_connector.ipynb b/docs/tutorials/05_torch_connector.ipynb index 6ee62c8a3..0a4abcb7f 100644 --- a/docs/tutorials/05_torch_connector.ipynb +++ b/docs/tutorials/05_torch_connector.ipynb @@ -321,6 +321,7 @@ ], "source": [ "# Evaluate model and compute accuracy\n", + "model1.eval()\n", "y_predict = []\n", "for x, y_target in zip(X, y):\n", " output = model1(Tensor(x))\n", @@ -499,6 +500,7 @@ ], "source": [ "# Evaluate model and compute accuracy\n", + "model2.eval()\n", "y_predict = []\n", "for x in X:\n", " output = model2(Tensor(x))\n", @@ -704,6 +706,7 @@ "plt.plot(X, y, \"bo\")\n", "\n", "# Plot fitted line\n", + "model3.eval()\n", "y_ = []\n", "for x in np.linspace(lb, ub):\n", " output = model3(Tensor([x]))\n",