Shortcuts

快速开始

本页用同一个模型和同一个数据集展示两种入口:配置文件适合保存、复用评测设置,也便于版本管理与结果复现,推荐使用;而 CLI 适合快速试跑。此处给出的示例将通过 OpenAI Responses API 评测 gpt-6-astra 在 GSM8K 的 64 条演示样本上的表现。

开始前,请先完成安装与环境准备并进入 OpenCompass 仓库根目录。随后打开 opencompass/configs/models/openai/gpt_6_astra.py,选择以下任一种方式配置 API 密钥。

对于临时的本地测试,可以直接将密钥传给模型配置:

from opencompass.models import OpenAISDKResponse

models = [
    dict(
        type=OpenAISDKResponse,
        abbr='gpt-6-astra-response',
        path='gpt-6-astra',
        key='your-api-key',
        # 其余参数不变
    )
]

更推荐将密钥保存到自定义环境变量中。例如,先在终端设置 OPENCOMPASS_API_KEY:

export OPENCOMPASS_API_KEY="your-api-key"

然后在模型配置中读取该环境变量:

import os
from opencompass.models import OpenAISDKResponse

models = [
    dict(
        type=OpenAISDKResponse,
        abbr='gpt-6-astra-response',
        path='gpt-6-astra',
        key=os.getenv('OPENCOMPASS_API_KEY'),
        # 其余参数不变
    )
]

本示例不需要本地 GPU,但运行前应确认当前环境和账号秘钥能够访问相应的 API 服务。首次运行还可能联网下载数据集。

路径一:使用配置文件

仓库已经提供 gpt-6-astra 模型配置与 GSM8K 演示数据集。新建 quick_start_eval.py:

# quick_start_eval.py
from mmengine.config import read_base

with read_base():
    from opencompass.configs.datasets.demo.demo_gsm8k_chat_gen import \
        gsm8k_datasets
    from opencompass.configs.models.openai.gpt_6_astra import \
        models as gpt6_astra_models

datasets = gsm8k_datasets
models = gpt6_astra_models

先通过添加 --dry-run 参数检查配置能否解析,以及推理任务如何划分。此参数不会真正执行模型推理:

opencompass quick_start_eval.py --dry-run

确认无误后运行完整流程:

opencompass quick_start_eval.py \
    --work-dir outputs/quick_start_config \
    --debug

--debug 会在当前进程中依次执行任务并直接打印日志到终端,适合第一次排错;正式批量评测通常不需要它。

路径二:直接使用 CLI

不创建配置文件也可以完成同一项评测。从 opencompass/configs/models 与 opencompass/configs/datasets 中获取配置文件名称(不含 .py 后缀),并作为 --models 和 --datasets 的参数值传入:

opencompass \
    --models gpt_6_astra \
    --datasets demo_gsm8k_chat_gen \
    --work-dir outputs/quick_start_cli \
    --debug

若要查找配置文件的名称,可运行配置搜索工具:

python tools/list_configs.py gpt_6_astra gsm8k

CLI 方式适合快速验证,但不适合需要精细控制的评测。需要组合多个模型、数据集,修改模型的并发与具体调用参数,以及自定义执行策略时,优先使用配置文件。

查看结果

每次运行会在 --work-dir 下创建时间戳目录,主要内容包括:

  • configs/:本次运行最终生效的配置快照;

  • predictions/:逐样本模型输出;

  • results/:评测器输出的指标与明细;

  • summary/:最终汇总结果,包括 CSV、Markdown 等格式。

如果执行失败,先检查终端或时间戳目录中的日志,并参阅“其他文档”章节的常见问题。完成本页后,建议阅读工作流与核心概念,再通过使用配置文件完成一次完整评测学习可复现的完整配置方式。

@沪ICP备2021009351号-23 OpenCompass Open Platform Service Agreement