flax
a3990ed1 - Switch to Optax optimizer and use TrainState.

Commit

5 years ago

Switch to Optax optimizer and use TrainState.

References

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

Author

bastings

bastings

Parents

Loading