Mojo package

checkpoint

APIs to save and load checkpoints for MAX graphs.

Modules

• ​metadata: Tools for reading metadata from a checkpoint file without completely loading all tensors.
• ​save_load: Defines functions to save/load tensors from a checkpoint file.
• ​tensor_dict: Structs used for maintaining a collection of tensors.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!
If you'd like to share more information, please report an issue on GitHub

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other