- Fixing bug that made graph scanner register repeated dense layers as regular dense layers.

google-deepmind / kfac-jax

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

Apache License 2.0

250 stars 23 forks source link

Closed copybara-service[bot] closed 2 weeks ago

copybara-service[bot] commented 2 weeks ago

Fixing bug that made graph scanner register repeated dense layers as regular dense layers.
Adding whitespace to improve readability.