- Adding overridable _post_param_update_processing method to experiment class in examples

google-deepmind / kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Apache License 2.0

235 stars 18 forks source link

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 2 months ago

Adding overridable _post_param_update_processing method to experiment class in examples