diff --git a/lectures/micrograd/micrograd_lecture_second_half_roughly.ipynb b/lectures/micrograd/micrograd_lecture_second_half_roughly.ipynb index 67a25a68..67a2cb41 100644 --- a/lectures/micrograd/micrograd_lecture_second_half_roughly.ipynb +++ b/lectures/micrograd/micrograd_lecture_second_half_roughly.ipynb @@ -1203,7 +1203,7 @@ " \n", " # forward pass\n", " ypred = [n(x) for x in xs]\n", - " loss = sum((yout - ygt)**2 for ygt, yout in zip(ys, ypred))\n", + " loss = sum((yout - ygt)**2 for ygt, yout in zip(ys, ypred)) / len(ys) # Video lecture calculates loss as SSE. Corrected to MSE here.\n", " \n", " # backward pass\n", " for p in n.parameters():\n",