Re-align docs with code example for mnist (#1364)
* Re-align docs with code example for mnist * Also fix dist_mnist and add context highlight
Showing
Please
register
or
sign in
to comment
* Re-align docs with code example for mnist * Also fix dist_mnist and add context highlight