flax
2a0a9e2c - allow passing a specific checkpoint path to restore_checkpoint.

Commit

5 years ago

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

References

Author

cghawthorne

Committer

a-googler

Parents