Skip to main content

module

save_load

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

Functions

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

Was this page helpful?