Skip to content

Commit 8cdc067

Browse files
noxthotmcabbott
andauthored
Fix mlp mnist explicit gradients (#385)
* style(mlp_mnist): remove trailing spaces * doc(mlp_mnist): add further user output about progress * feat(mlp_mnist): use explicit gradients * feat(mlp_mnist): use same loss function during train process and reporting * build(mlp_mnist): compat Flux = 0.13.9 and MLDatasets = 0.7; update Manifest * doc(README): reflect update of mlp_mnist to use explicit gradients * chore(mlp_mnist): ADAM -> Adam Co-authored-by: Michael Abbott <[email protected]> * chore(mlp_mnist): define and use const loss; aggregate loss within loss_and_accuracy by sum (instead of mean) to restore original behaviour Co-authored-by: Michael Abbott <[email protected]>
1 parent ed6cabb commit 8cdc067

File tree

4 files changed

+324
-262
lines changed

4 files changed

+324
-262
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ Flux v0.13 is the latest right now, marked with ☀️; models upgraded to use
6565

6666
**Vision**
6767
* MNIST
68-
* [Simple multi-layer perceptron](vision/mlp_mnist) ☀️ v0.13
68+
* [Simple multi-layer perceptron](vision/mlp_mnist) ☀️ v0.13 +
6969
* [Simple ConvNet (LeNet)](vision/conv_mnist) ☀️ v0.13
7070
* [Variational Auto-Encoder](vision/vae_mnist) ☀️ v0.13 +
7171
* [Deep Convolutional Generative Adversarial Networks](vision/dcgan_mnist) ☀️ v0.13 +

0 commit comments

Comments
 (0)