Skip to content

Fixed broken checkpoint API names and add composite model+data example (#938)#1231

Open
vasilybib1 wants to merge 1 commit intogoogle:mainfrom
vasilybib1:fix/documentation
Open

Fixed broken checkpoint API names and add composite model+data example (#938)#1231
vasilybib1 wants to merge 1 commit intogoogle:mainfrom
vasilybib1:fix/documentation

Conversation

@vasilybib1
Copy link

@vasilybib1 vasilybib1 commented Feb 27, 2026

  • Fix grain.GrainCheckpointSave/Restoregrain.checkpoint.CheckpointSave/Restore
  • Add "Checkpointing alongside model state" section showing ocp.args.Composite pattern for saving model and data iterator state together

Fixes #938


📚 Documentation preview 📚: https://google-grain--1231.org.readthedocs.build/

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add dataloader checkpointing example to docs

1 participant