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