cerebras.modelzoo.tools.convert_checkpoint#

Functions

convert_checkpoint

convert_checkpoint_from_file

convert_config

convert_config_from_file

diff_checkpoints

Compare state dictionaries of two checkpoints (left and right).

diff_checkpoints_from_file

Compare two checkpoints (left and right).

get_model_converter

Classes

CheckpointConverterCLI