一道 GPT-2 参数计算题:仅 embed 与 unembed 用多少参数?

Sauers_ · x · 2026-10-17

作者 Sauers 出了一道关于 GPT-2 参数结构的技术小测验:GPT-2 在网络开头将词表中的每个 token 嵌入为长度 E 的向量(embedding 矩阵 V×E),在结尾又将最终 hidden state 反投影回 token logits(unembedding,通常与 embedding 权重共享)。只计这两步,可训练参数量为 2×V×E。以 GPT-2 最常见的 V=50257、E=768 计算,约为 7720 万参数;且由于权重绑定,实际独立参数只有 V×E 一份。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →