【免费下载链接】GSM8K-in256-bs1
GSM8K-in256-bs1是面向 vLLM-Ascend(Ascend NPU 上的 vLLM 推理引擎)的一个轻量级GSM8K 数学推理评测数据集,专门用于在固定 256 token 输入长度、batch size 为 1 的工况下,对大模型的数学推理能力与推理引擎性能做快速验证。整个仓库只有 3 个文件,却能帮你 5 分钟跑通一次评测流程。
一、它到底是什么?📦
GSM8K 是业界公认的小学数学应用题基准(8.5K 道),常用于检验大模型的**数学推理(Math Reasoning)**能力。而GSM8K-in256-bs1则是它在 Ascend 场景下的一个"精准调参版"切片:
- in256:输入(input)被构造为约 256 个 token 的定长请求;
- bs1:batch size = 1,即单请求工况;
- Minimax:文件命名表明该数据是为 MiniMax 系列模型在 vLLM-Ascend 上跑批时生成的。
这种"定长输入 + 单批次"的组合,是推理引擎性能验证(首 token 时延 TTFT、解码吞吐、显存/NPU 占用)最常用的测试形态。
仓库文件一览
| 文件 | 大小 | 作用 |
|---|---|---|
| GSM8K-in256-bs1-Minimax.json | 4 KB | JSON 格式,单条评测请求 |
| GSM8K-in256-bs1-Minimax.jsonl | 4 KB | JSONL 格式,内容与上面完全一致 |
| README.md | 4 KB | 数据集卡片(许可证 Apache 2.0 等元信息) |
二、命名拆解:in256 与 bs1 各指什么?
| 片段 | 含义 | 评测意义 |
|---|---|---|
GSM8K | 基准来源 | 数学应用题,考察多步算术推理 |
in256 | 输入长度 ≈ 256 tokens | 模拟"中短 prompt"工况,结果可横向对比 |
bs1 | 批大小 = 1 | 单请求压测,聚焦单路延迟与 prefill 阶段表现 |
Minimax | 目标模型 | 配套 MiniMax 模型在 Ascend NPU 上的验证 |
三、数据格式详解:一条 GSM8K 测试题长什么样?
打开 GSM8K-in256-bs1-Minimax.json,整个文件就是一个 JSON 对象,只有两个字段:
{ "question": "Janet's ducks lay 16 eggs per day. ... x x x x", "answer": "none" }question:经典的 GSM8K 原题——"Janet 的鸭子每天下 16 个蛋,她吃 3 个、烤 4 个麦芬,剩下的按每个 2 美元卖,每天赚多少?"(正确答案 18 美元)。为了让输入稳定落在 256 token 附近,题目被重复拼接了 4 次,末尾补了x x x x占位符;answer为none:说明该数据集的定位不是"答题判分",而是固定输入长度的性能/精度冒烟测试——引擎只关心能否稳定输出,不关心答案对错。
.jsonl版本 GSM8K-in256-bs1-Minimax.jsonl 与.json内容一致(每行一个 JSON 对象),方便按行流式读取,适配vllm bench一类批量压测工具。
💡 小提示:GSM8K 是标准多步算术题,4 次重复会稀释语义强度,因此把它当"256 token 定长探针"看,比当严格能力基准看更合理。
四、典型使用场景:在 Ascend NPU 上验证 vLLM-Ascend
这个数据集的典型工作流是:
- 用 vLLM-Ascend 拉起目标模型(如 MiniMax 系列量化版,W8A8/W4A8 等);
- 以
bs1+ 256 token 输入发起单请求; - 观察 TTFT(首 token 时延)、生成吞吐与 NPU 利用率是否符合预期;
- 换模型 / 换量化格式重复对比,定位性能差异。
它和 vLLM-Ascend 仓库族中的其他量化模型数据集(如 Qwen3-8B-W4A8、GLM-4.7-W8A8C8)配合使用,可以组成一套"固定工况矩阵":同一输入形态,横向比较不同模型与量化方案的推理表现。
五、最快获取方法:一行命令 clone
git clone https://gitcode.com/hf_mirrors/vllm-ascend/GSM8K-in256-bs1完成后目录内即为上述 3 个文件。若你使用 ModelScope SDK 管理数据集,也可以参照 README.md 中的snapshot_download('vllm-ascend/GSM8K-in256-bs1')方式获取。
六、常见问题 FAQ ❓
Q1:为什么只有一道题?因为它的角色是"工况探针"而非完整基准。固定 1 条 256 token 请求,才能把变量收敛到引擎与模型本身。
Q2:answer为什么是none?表示不做答案判分。评测目标是推理性能与输出稳定性,而非题目对错。
Q3:.json和.jsonl该用哪个?两者内容相同。脚本按文件整体加载用.json,按行流式读取或喂给压测工具用.jsonl。
Q4:能直接当 GSM8K 完整基准用吗?不能。完整 GSM8K 含数千道题;这里是单题定长切片,适合冒烟测试与性能基线对比。
写在最后
GSM8K-in256-bs1的价值不在题目数量,而在"标准化工况"这四个字:256 token 输入 + batch size 1,让每一次 vLLM-Ascend 上的模型验证都有可复现、可对比的起点。对正在 Ascend NPU 上部署大模型的同学来说,这就是一个开箱即用的数学推理评测最小单元。
【免费下载链接】GSM8K-in256-bs1
相关推荐
TinyZero数学推理能力深度评测:GSM8K和数学数据集表现分析
TinyZero数学推理能力深度评测:GSM8K和数学数据集表现分析 TinyZero作为DeepSeek R1 Zero的简洁复现版本,在数学推理任务上展现出
人工智能大模型强化学习推理模型微调分布式训练Qwen3-30B-A3B数学推理能力评测:GSM8K与MATH数据集表现分析
Qwen3 30B A3B数学推理能力评测:GSM8K与MATH数据集表现分析 引言:大模型数学推理的痛点与突破 你是否在使用大语言模型(Large Langu
大模型基础模型人工智能SSD vs vLLM vs SGLang:LLM推理引擎基准测试完全指南(含4大数据集)
SSD vs vLLM vs SGLang:LLM推理引擎基准测试完全指南(含4大数据集) 本文带你完整了解 SSD(Speculative Speculati
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考