格式化检查点文件¶

class torchtune.training.FormattedCheckpointFiles(filename_format: str, max_filename: str)[source]¶

这个类提供了一种更简洁的方式来表示格式为file_{i}_of_{n_files}.pth的文件名列表。

Parameters:

filename_format (str) – 文件名的格式字符串。必须包含两个占位符，例如 file_{}_of_{}.pth。
max_filename (str) – 列表中的最大文件名。应该是一个整数的字符串表示，可能带有前导零。

build_checkpoint_filenames()[source]¶

从文件名格式和最大文件名构建检查点文件名的列表。

Returns:: 检查点文件名列表。
Return type:: 列表[str]

示例

>>> # Example usage
>>> f = FormattedCheckpointFiles(filename_format="file_{}_of_{}.safetensors", max_filename="00003")
>>> f.build_checkpoint_filenames()
>>> ['file_00001_of_00003.safetensors', 'file_00002_of_00003.safetensors', 'file_00003_of_00003.safetensors']