HOWTO: Loading MNIST from torchvision

[rolandgvc]

Useful small howto on loading torchvision datasets.

[avital]

Nice! Tests are failing (https://github.com/google/flax/pull/325/checks?check_run_id=770571323#step:8:202) because of the torchvision dependency. I think you need to update examples/requirements.txt.

[avital]

Hmm maybe you also need to update tests_require in setup.py

[avital]

@rolandgvc Looks like tests are still failing on GitHub actions, it's not super clear to me why. Can you try running tests locally? Do they pass there?

[avital]

We've redone our HOWTO system. Here's an example of the new format:

Rendered in ReadTheDocs: https://flax.readthedocs.io/en/latest/howtos/ensembling.html
Document source: https://github.com/google/flax/blob/master/docs/howtos/ensembling.rst (see the :testsetup, :testcode and :codediff directives)

We'd happy take pull requests revamping this HOWTO to the new system.

[marcvanzee]

Closing this and replacing it with #1853