flax
995bc01f - Merge pull request #2071 from andsteing:mnist

Commit

3 years ago

Merge pull request #2071 from andsteing:mnist PiperOrigin-RevId: 444784231

References

#2071 - Makes annotated_mnist use Optax's xent loss.

Author

a-googler

a-googler

Parents

Loading