Skip to content

Commit

Permalink
Add user-facing CheckpointMetadata classes and Internal classes for IO.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 691962281
  • Loading branch information
BlaziusMaximus authored and Orbax Authors committed Nov 6, 2024
1 parent e47257e commit 5831f7f
Show file tree
Hide file tree
Showing 15 changed files with 847 additions and 386 deletions.
8 changes: 8 additions & 0 deletions checkpoint/CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,19 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0

## [Unreleased]

### Added
- Add `RootMetadata` and `StepMetadata` classes as ways for the user to
interface with checkpoint metadata at various levels.
- Add `root_metadata_serialization`, and `step_metadata_io` modules that contain utilities
to perform de/serialization for `RootMetadata` and `StepMetadata`.

### Changed
- Create `Composite` class, which `CompositeArgs` now subclasses.
- Move `type_handlers` to `_src/serialization`
- Add notes to Barrier error `XlaRuntimeError(DEADLINE_EXCEEDED)` with
actionable info.
- Rename `CheckpointMetadataStore` to `MetadataStore`, and change methods to
accept and return metadata as dictionaries.

## [0.8.0] - 2024-10-29

Expand Down
Loading

0 comments on commit 5831f7f

Please sign in to comment.