跳转至

checkpoint/ 子模块 — 检查点管理

概述

checkpoint/ 子模块管理训练过程中模型状态的保存和恢复。支持 FSDP 和 Megatron 两种训练策略的检查点管理,以及自动轮转(删除旧检查点)。

文件列表

文件 说明
__init__.py 模块入口
checkpoint_manager.py 基础检查点管理器
checkpoint_handler.py 保存/加载处理器接口
fsdp_checkpoint_manager.py FSDP 策略检查点管理
megatron_checkpoint_manager.py Megatron 策略检查点管理