We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e6405b9 commit 9ad2e9eCopy full SHA for 9ad2e9e
site/en/guide/core/mlp_core.ipynb
@@ -400,8 +400,8 @@
400
" if not self.built:\n",
401
" # Infer the input dimension based on first call\n",
402
" self.in_dim = x.shape[1]\n",
403
- " # Initialize the weights and biases using Xavier scheme\n",
404
- " self.w = tf.Variable(xavier_init(shape=(self.in_dim, self.out_dim)))\n",
+ " # Initialize the weights and biases\n",
+ " self.w = tf.Variable(self.weight_init(shape=(self.in_dim, self.out_dim)))\n",
405
" self.b = tf.Variable(tf.zeros(shape=(self.out_dim,)))\n",
406
" self.built = True\n",
407
" # Compute the forward pass\n",
0 commit comments