Misc improvements in resnet_flax example.

* Better data augmentation, leading to 88% accuracy (from 70%) * Plots showing the data augmentation in action. * Options use the same format as distributed training examples. * Changed solver from Adam to SGD for better accuracy.
google · Apr 14, 2023 · fff693f · fff693f
1 parent 4edd8ac
commit fff693f
Show file tree

Hide file tree

Showing 2 changed files with 942 additions and 420 deletions.
diff --git a/docs/notebooks/deep_learning/resnet_flax.ipynb b/docs/notebooks/deep_learning/resnet_flax.ipynb