diff --git a/tensorflow_v2/notebooks/3_NeuralNetworks/autoencoder.ipynb b/tensorflow_v2/notebooks/3_NeuralNetworks/autoencoder.ipynb index b7c22279..67fe2014 100644 --- a/tensorflow_v2/notebooks/3_NeuralNetworks/autoencoder.ipynb +++ b/tensorflow_v2/notebooks/3_NeuralNetworks/autoencoder.ipynb @@ -178,7 +178,7 @@ " loss = mean_square(reconstructed_image, x)\n", "\n", " # Variables to update, i.e. trainable variables.\n", - " trainable_variables = weights.values() + biases.values()\n", + " trainable_variables = list(weights.values()) + list(biases.values())\n", " \n", " # Compute gradients.\n", " gradients = g.gradient(loss, trainable_variables)\n",