checkpoint_handler.py — 检查点保存/加载处理器¶
文件路径: verl/utils/checkpoint/checkpoint_handler.py
文件概述¶
定义检查点保存和加载的处理器接口。不同的模型组件(Actor、Critic、优化器等)有不同的保存逻辑,通过 handler 模式统一管理。
核心接口¶
class CheckpointHandler:
def save(self, path, **kwargs):
"""保存到指定路径"""
raise NotImplementedError
def load(self, path, **kwargs):
"""从指定路径加载"""
raise NotImplementedError
与其他模块的关系¶
- 被 checkpoint_manager 调用
- 具体实现由 FSDP/Megatron 的 checkpoint manager 提供
小结¶
检查点保存/加载的抽象接口。