flax
e298d6c1 - Makes annotated_mnist use Optax's xent loss.

Commit

3 years ago

Makes annotated_mnist use Optax's xent loss.

References

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

Author

andsteing

andsteing

Parents

Loading