Skip to main content

Mojo module

save_load

Defines functions to save/load tensors from a checkpoint file.

Functions

  • load: Reads tensors from saved checkpoint file.
  • save: Saves a collection of tensors to a file.