flax
1768cb53 - allow passing a specific checkpoint path to restore_checkpoint.

Commit

5 years ago

allow passing a specific checkpoint path to restore_checkpoint. PiperOrigin-RevId: 353786108

References

Author

cghawthorne

Committer

a-googler

Parents