Skip to content

Commit

Permalink
Merge pull request #4837 from mdanatg/master
Browse files Browse the repository at this point in the history
 Update the training schedule for better convergence
  • Loading branch information
MarkDaoust committed Jul 19, 2018
2 parents 6adf454 + e6f7756 commit 71c196c
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions samples/core/guide/autograph.ipynb
Expand Up @@ -740,7 +740,7 @@
"@autograph.convert(recursive=True)\n",
"def train(train_ds, test_ds, hp):\n",
" m = mlp_model((28 * 28,))\n",
" opt = tf.train.MomentumOptimizer(hp.learning_rate, 0.9)\n",
" opt = tf.train.AdamOptimizer(hp.learning_rate)\n",
" \n",
" # We'd like to save our losses to a list. In order for AutoGraph\n",
" # to convert these lists into their graph equivalent,\n",
Expand Down Expand Up @@ -802,7 +802,7 @@
"source": [
"with tf.Graph().as_default() as g:\n",
" hp = tf.contrib.training.HParams(\n",
" learning_rate=0.05,\n",
" learning_rate=0.005,\n",
" max_steps=500,\n",
" )\n",
" train_ds = setup_mnist_data(True, 50)\n",
Expand Down Expand Up @@ -837,4 +837,4 @@
"outputs": []
}
]
}
}

0 comments on commit 71c196c

Please sign in to comment.