跳到正文
原文
Google AI:DEV 作者专属(RSS)· S Aaron Dennis·· 9 小时前AI 评分57

用 PyTorch 从零构建并训练 124M 参数的 GPT-2 风格 decoder-only Transformer

Demystifying LLMs: Building a 124M-Parameter Decoder-Only Transformer in PyTorch

AI 导读

作者用 PyTorch 从零实现并训练了一个 124M 参数的 GPT-2 风格 decoder-only Transformer,配置为 768 维嵌入、12 头注意力、12 层、词表 50,257、最大序列长度 512。

来源:Google AI:DEV 作者专属(RSS) · dev.to