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

Merged

Makes annotated_mnist use Optax's xent loss. #2071

copybara-service merged 2 commits into google:main from andsteing:mnist

Makes annotated_mnist use Optax's xent loss.

e298d6c1

andsteing requested a review from

marcvanzee 4 years ago

andsteing assigned

andsteing 4 years ago

andsteing added Priority: P1 - soon

marcvanzee approved these changes on 2022-04-26

andsteing requested a review from

marcvanzee 4 years ago

marcvanzee approved these changes on 2022-04-26

ayaka14732 approved these changes on 2022-04-26

Makes GPU info shorter (-L flag).

5a7a345f

andsteing added pull ready

copybara-service merged 995bc01f into main 4 years ago

Reviewers

ayaka14732

marcvanzee

Assignees

andsteing

Labels

Priority: P1 - soon pull ready

Milestone

No milestone