跳转至

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 提供

小结

检查点保存/加载的抽象接口。