HPUCheckpointIO¶

class lightning.pytorch.plugins.io.HPUCheckpointIO[source]¶

CheckpointIO to save checkpoints for HPU training strategies.

Warning

This is an experimental feature.

save_checkpoint(checkpoint, path, storage_options=None)[source]¶

Save model/training states as a checkpoint file through state-dump and file-write.

Parameters

checkpoint¶ (Dict[str, Any]) – dict containing model and trainer state
path¶ (Union[str, Path]) – write-target path
storage_options¶ (Optional[Any]) – not used in XLACheckpointIO.save_checkpoint

Raises

TypeError – If storage_options arg is passed in

Return type

None