Add comment about the reshape operation from the training_step function

main
Aurélien Geron 2020-03-12 22:51:36 +13:00
parent 49715d4b74
commit cd4e2e1313
1 changed files with 3 additions and 1 deletions

View File

@ -1376,7 +1376,9 @@
"cell_type": "markdown",
"metadata": {},
"source": [
"Lastly, let's create a function that will sample some experiences from the replay memory and perform a training step:"
"Lastly, let's create a function that will sample some experiences from the replay memory and perform a training step:\n",
"\n",
"**Note**: the first 3 releases of the 2nd edition were missing the `reshape()` operation which converts `target_Q_values` to a column vector (this is required by the `loss_fn()`)."
]
},
{