CheckpointSaveConfig

Module: fast_llm.engine.checkpoint.config

Inherits from: CheckpointSaveMetadataConfig, CheckpointPathConfigBase, CheckpointStateSaveConfigBase

Fields

format — core

Type: type    Default: FastLLMCheckpointFormat

Format of the checkpoint.

path — core

Type: Path or None    Default: None

Location of the checkpoint.

data_type — feature

Type: DataType or None    Default: None

Data type to save the checkpoint.

model_weights — feature

Type: bool    Default: True

Save the model weights.

optimizer_state — feature

Type: bool    Default: None

Save the optimizer state. Default: save if supported by the format.

parameters_per_file — feature

Type: int    Default: 4_294_967_296

Limit the number of parameters saved in each file.

timeout — optional

Type: float or None    Default: None

Custom timeout for lengthy operations.