From 74b2a2783d137b183b7c7016eb9fa3f93895e572 Mon Sep 17 00:00:00 2001 From: Soban Malik <57958515+sobanmalik@users.noreply.github.com> Date: Fri, 22 Nov 2019 02:08:09 +0530 Subject: [PATCH] GradientTape change --- .../notebooks/3_NeuralNetworks/neural_network_raw.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb b/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb index bbec2f13..5ac19d89 100644 --- a/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb +++ b/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb @@ -180,7 +180,7 @@ " loss = cross_entropy(pred, y)\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",