Move to use Optax optimizer. - SemanticDiff

Commit

4 years ago

Move to use Optax optimizer.

References

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

Author

bastings

bastings

Parents

Loading