FormattedCheckpointFiles

类 torchtune.training.FormattedCheckpointFiles(filename_format: str, max_filename: str)

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

参数：

• filename_format (str) – 文件名的格式字符串。必须恰好包含两个占位符，例如 file_{}_of_{}.pth。

• max_filename (str) – 列表中的最大文件名。应该是一个整数的字符串表示，可能带有前导零。

build_checkpoint_filenames()

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

返回值：

检查点文件名列表。

返回类型：

List[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']