made initial weights higher-variance

main
arodiss 2018-02-10 19:24:56 +02:00
parent a4c4c714eb
commit 3a72a2b4aa
1 changed files with 2 additions and 2 deletions

View File

@ -537,7 +537,7 @@
"cell_type": "markdown",
"metadata": {},
"source": [
"Okay, let's start by creating a trainable variable of shape (1, 1152, 10, 16, 8) that will hold all the transformation matrices. The first dimension of size 1 will make this array easy to tile. We initialize this variable randomly using a normal distribution with a standard deviation to 0.01."
"Okay, let's start by creating a trainable variable of shape (1, 1152, 10, 16, 8) that will hold all the transformation matrices. The first dimension of size 1 will make this array easy to tile. We initialize this variable randomly using a normal distribution with a standard deviation to 0.1."
]
},
{
@ -546,7 +546,7 @@
"metadata": {},
"outputs": [],
"source": [
"init_sigma = 0.01\n",
"init_sigma = 0.1\n",
"\n",
"W_init = tf.random_normal(\n",
" shape=(1, caps1_n_caps, caps2_n_caps, caps2_n_dims, caps1_n_dims),\n",