checkpoint/ 子模块 — 检查点管理¶
概述¶
checkpoint/ 子模块管理训练过程中模型状态的保存和恢复。支持 FSDP 和 Megatron 两种训练策略的检查点管理,以及自动轮转(删除旧检查点)。
文件列表¶
| 文件 | 说明 |
|---|---|
__init__.py |
模块入口 |
checkpoint_manager.py |
基础检查点管理器 |
checkpoint_handler.py |
保存/加载处理器接口 |
fsdp_checkpoint_manager.py |
FSDP 策略检查点管理 |
megatron_checkpoint_manager.py |
Megatron 策略检查点管理 |