Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Remove usages of orbax_utils.save_args_from_target, as this function …
…does nothing (it used to control a checkpointing behavior that has since been optimized away). PiperOrigin-RevId: 716314210
- Loading branch information