English(EN) Training a 210M text-to-image DiT from scratch on one GPU: what I measured [P]

研究人员从头开始在单GPU上训练2.1亿参数文本到图像DiT模型

作者 PulseAugur 编辑部 · [2 个来源] · 2026-09-10 13:18

一位研究人员从头开始训练了一个拥有2.1亿参数的文本到图像扩散Transformer模型，并在单台GPU上花费了3.5天完成。该实验的关键发现包括：学习到的空注意力槽成为交叉注意力质量的主要汇聚点；流匹配损失作为健康信号而非直接质量指标；训练时间步长偏移比简单地将训练步数加倍更具影响力。 AI

影响为扩散模型和注意力机制的高效训练提供了见解，可能为未来的模型开发提供信息。

排序理由研究论文，详细介绍了扩散Transformer模型的训练方法和发现。

AI 生成摘要 · Google Gemini · 来自 2 个来源。我们如何撰写摘要 →

本文如何被排名

Signal score: 0 / 100; Composite score across the factors below. Higher = stronger signal that this story matters right now.
Newsworthiness bucket: Research; 研究论文，详细介绍了扩散Transformer模型的训练方法和发现。
Source corroboration: 2 independent sources; Multiple independent publishers reporting the same story raises confidence that it's real and newsworthy.
Topics: model release, infra; Editorial topic classification. Feeds into how the story surfaces on /topic/<slug> hub pages and into the per-entity coverage mix.
AI-industry relevance: High; Clearly on-topic for AI-industry coverage.
Story freshness: 45 days old; Aged out of breaking-news scoring windows; ranking reflects the durable signal from the full source set.

完整方法见我们的编辑标准。

r/MachineLearning TIER_1 English(EN) · /u/IvanMikhnenkov · 2026-09-11 13:00

从头开始在一台GPU上训练一个2.1亿参数的text-to-image DiT模型：我的测量结果 [P]

<table> <tr><td> <a href="https://www.reddit.com/r/MachineLearning/comments/1wdfmvq/training_a_210m_texttoimage_dit_from_scratch_on/"> <img alt="Training a 210M text-to-image DiT from scratch on one GPU: what I measured [P]" src="https://preview.redd.it/7vrhqzf32woh1.png?width=14…
r/StableDiffusion TIER_2 English(EN) · /u/IvanMikhnenkov · 2026-09-10 13:18

我用一块GPU从头开始训练了一个210M的文本到图像扩散Transformer，耗时3.5天

<table> <tr><td> <a href="https://www.reddit.com/r/StableDiffusion/comments/1wciz7m/i_trained_a_210m_texttoimage_diffusion/"> <img alt="I trained a 210M text-to-image diffusion transformer from scratch on one GPU in 3.5 days" src="https://external-preview.redd.it/NHhueGM0NTExcG9o…