flax
48b5707a - Merge pull request #1306 from bastings:sst2_optax

Commit

4 years ago

Merge pull request #1306 from bastings:sst2_optax PiperOrigin-RevId: 372564394

References

#1306 - sst2 example: Switch to Optax optimizer and use TrainState.

Author

a-googler

a-googler

Parents

Loading