flax
6bab1bf9 - Use Optax optimizer for ppo example

Commit

4 years ago

Use Optax optimizer for ppo example - Use TrainState - use apply_fn inline with train_state pattern

References

Author

jheek

Committer

jheek

Parents