diff --git a/extra_capsnets.ipynb b/extra_capsnets.ipynb index 20e5dd1..1569b4a 100644 --- a/extra_capsnets.ipynb +++ b/extra_capsnets.ipynb @@ -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",