flax
9ff5feb3 - Fix LARS optimizer for zero param initialization.

Commit

5 years ago

Fix LARS optimizer for zero param initialization. PiperOrigin-RevId: 331122839

References

#459 - Fix LARS optimizer for zero param initialization.

Author

a-googler

a-googler

Committer

a-googler

a-googler

Parents

Loading