modalities.checkpointing package

Subpackages

• modalities.checkpointing.fsdp package
  • Submodules
  • modalities.checkpointing.fsdp.fsdp_checkpoint_loading module
  • modalities.checkpointing.fsdp.fsdp_checkpoint_saving module
  • Module contents
• modalities.checkpointing.torch package
  • Submodules
  • modalities.checkpointing.torch.torch_checkpoint_loading module
  • modalities.checkpointing.torch.torch_checkpoint_saving module
  • Module contents

Submodules

modalities.checkpointing.checkpoint_conversion module

modalities.checkpointing.checkpoint_loading module

modalities.checkpointing.checkpoint_saving module

modalities.checkpointing.checkpoint_saving_execution module

modalities.checkpointing.checkpoint_saving_instruction module

class modalities.checkpointing.checkpoint_saving_instruction.CheckpointingInstruction(save_current=False, checkpoints_to_delete=<factory>)

Bases: object

Represents a checkpointing instruction (i.e., saving and deleting).

Attributes:

save_current (bool): Indicates whether to save the current checkpoint. checkpoints_to_delete (list[TrainingProgress]): List of checkpoint IDs to delete.

Parameters:

• save_current (bool)

• checkpoints_to_delete (list[TrainingProgress])

checkpoints_to_delete: list[TrainingProgress]

save_current: bool = False

modalities.checkpointing.checkpoint_saving_strategies module

Module contents