ソースを参照

Update convolutional_network_raw.ipynb (#366)

lkdmttg7 5 年 前
コミット
39d9d0efa1

+ 1 - 1
tensorflow_v2/notebooks/3_NeuralNetworks/convolutional_network_raw.ipynb

@@ -217,7 +217,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",