T learn.traeai

官方视频 · Sebastian Raschka

Loading a base model, text generation, and KV caching

点击后加载第三方播放器,视频由原平台提供。播放取决于网络与原平台的嵌入设置;若无法播放,可使用上方官方观看链接。下方中文导读和练习由本站独立编写。

第 2 节 · 本站原创中文课程 · 建议 30 分钟

第二集:跑通生成与 KV cache

学完能做:建立固定输入的生成基线,分别验证正确性与速度。

建议时间仅指中文导读与练习,不含官方视频和本地实验。代码请在自己的 Python / PyTorch 环境运行。

先写基线契约

在实验档案里固定题目、模型权重、生成上限和停止条件。保存原始输出,不先凭肉眼选一条‘最好看’的回答。后续优化都复用这组条件,让输出差异有依据可查。

缓存要验证边界

KV cache 的练习重点是复用历史注意力计算,同时正确管理新 token 的位置和缓存状态。设计同一提示的缓存开关对照,再连续输入两道无关题,检查上一题的状态有没有泄漏进下一题。

计时包含哪些工作?

分别记录模型加载、首次运行与预热后的生成时间。比较 torch.compile 时把编译时间单列,固定生成 token 数,并对异步设备做必要同步。若设备不支持某配置,报告不支持原因,不把该项计为零耗时。

这是本站独立编写的讲解与示例,并非官方课程逐字翻译。理解之后,请完成下面的练习、自测和笔记。

参考来源:Sebastian Raschka · Building Reasoning Models from the Ground Up

动手,把理解变成成果

本节练习

正在读取本机草稿…

用官方第 2 章代码设计并运行一组缓存开关对照。至少记录两道题的原始回答、生成长度、停止原因与预热后耗时;编译实验可标待测。

65 / 6000 字。本页不执行代码;可下载后在本地运行,或用 Agent 获取定性建议。

给我一个提示

先保证缓存前后在固定解码条件下结果相符,再比较性能。

查看参考思路(本站预写,非实时 Agent)

固定贪心解码和 token 上限,对两道题各运行缓存开关配置,每题先清空缓存。保存输出并检查一致性;将加载、首轮和预热后耗时分别列出。未测 torch.compile 就标待测。

人工自评:逐项核对自己的作品

这些勾选由你确认,不是自动测试结果。遇到待测项,请在作品里明确注明。

理解自测

加缓存后速度更快,但同一输入的输出异常变化,应先做什么?

自己的学习笔记

0 / 3000 字 · 仅保存在此浏览器

围绕当前章节学习

Agent 答疑与作品评价

正在检查连接…

本站讲解、提示、参考思路和自测可独立使用。连接模型服务后,才能实时提问或评价;这里不会把预写内容伪装为 Agent 回答。

走完学习闭环

完成本节

保留练习、勾选人工自评并通过自测后,可记录完成。修改作品或答案会重新打开本节。

课程进度:0 / 8 · 本机记录,不是外部官方证书

← 上一节:第一集:准备环境,建立实验档案