cerebras.modelzoo.common.checkpoint_utils#

Various utility functions related to checkpoint saving and loading.

Functions

get_all_checkpoints

Return the path to all available checkpoints

get_latest_checkpoint

Get the path to the checkpoint with the highest global step

Classes

CkptInfo

Class to manage checkpoints created within one run.