orbax icon indicating copy to clipboard operation
orbax copied to clipboard
verified publisher icon google

Add OrbaxCheckpointManager in Pax which aligns with Orbax APIs. A few overrides are needed for full compatibility with Pax checkpoint format and extra features.

Open copybara-service[bot] opened this issue 3 years ago • 0 comments

Add OrbaxCheckpointManager in Pax which aligns with Orbax APIs. A few overrides are needed for full compatibility with Pax checkpoint format and extra features.

copybara-service[bot] avatar Aug 23 '22 17:08 copybara-service[bot]