跳转至

__init__.py — 模块入口

文件路径: verl/utils/__init__.py

文件概述

模块初始化文件,定义了 verl.utils 的公共 API。

内容

from . import config, tokenizer
from .config import omega_conf_to_dataclass, validate_config
from .groupwise import as_torch_index, group_mean_std
from .tokenizer import hf_processor, hf_tokenizer, normalize_token_ids

__all__ = (
    tokenizer.__all__
    + config.__all__
    + ["hf_processor", "hf_tokenizer", "normalize_token_ids",
       "omega_conf_to_dataclass", "validate_config"]
    + ["as_torch_index", "group_mean_std"]
)

导出的公共 API

API 来源 说明
hf_tokenizer() tokenizer.py 加载 HF tokenizer
hf_processor() tokenizer.py 加载 HF processor
normalize_token_ids() tokenizer.py 标准化 token ID
omega_conf_to_dataclass() config.py 配置转换
validate_config() config.py 配置校验
group_mean_std() groupwise.py 分组均值/标准差
as_torch_index() groupwise.py 索引转换

小结

__init__.py 只导出了最常用的几个函数,其他工具需要从子模块直接导入。