{"id":23284342,"url":"https://github.com/gavinhome/llm-from-zero","last_synced_at":"2025-04-06T15:15:35.339Z","repository":{"id":267910323,"uuid":"902728831","full_name":"GavinHome/llm-from-zero","owner":"GavinHome","description":"从零开始构建一个大型语言模型（LLM, Large Language Model）","archived":false,"fork":false,"pushed_at":"2024-12-19T05:36:07.000Z","size":512,"stargazers_count":2,"open_issues_count":0,"forks_count":0,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-02-12T21:16:51.080Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Jupyter Notebook","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/GavinHome.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2024-12-13T06:38:11.000Z","updated_at":"2025-01-11T04:22:29.000Z","dependencies_parsed_at":"2024-12-13T07:30:55.617Z","dependency_job_id":"e2dca79c-f003-4f47-990f-6a3aa9ed8a0f","html_url":"https://github.com/GavinHome/llm-from-zero","commit_stats":null,"previous_names":["gavinhome/llm-from-zero"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/GavinHome%2Fllm-from-zero","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/GavinHome%2Fllm-from-zero/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/GavinHome%2Fllm-from-zero/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/GavinHome%2Fllm-from-zero/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/GavinHome","download_url":"https://codeload.github.com/GavinHome/llm-from-zero/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":247500477,"owners_count":20948880,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2024-12-20T01:40:07.914Z","updated_at":"2025-04-06T15:15:35.307Z","avatar_url":"https://github.com/GavinHome.png","language":"Jupyter Notebook","funding_links":[],"categories":[],"sub_categories":[],"readme":"# 从零开始构建一个大型语言模型\n本文是使用PyTorch从0开始逐步实现一个类似ChatGPT那样的大型语言模型（以下简称LLM），步骤大体分为如下：\n  - 数据准备与预处理\n  - 模型架构与实现\n  - 模型训练与评估\n  - 文本生成与微调\n\n## 前言\n\n如果你让ChatGPT继续输出“每一次努力都让你感动”后面的内容，ChatGPT会自动延续后续内容并输出，我第一次看到就对此产生了好奇。本文是实现一个类似这样的类GPT模型，且针对文本数据，力求尽量不出现公式，尽量减少专业术语，使用简单清晰的词语进行解释，并给出代码实现。初步可能为了更好的理解，实现一个简单的版本，它可能输出的并不理想，但最终会调整为适合现有大模型的复杂实现思路。故你可以将本文的目的理解为新手入门LLM通识训练。\n废话不多说，现在开始！\n\n首先，我们要先明确我们此行的目的：给一段输入“每一次努力都让你感动”，经过模型的处理加工后，输出后续的内容“未来的精彩由此慢慢绽放”。用程序化的语言可以这么表达：输入input，经过model的处理，输出output。一般情况下，模型是依据历史文本来生成固定长度的内容，不可能无限制的生成，所以还需要限定输入长度和输出长度，由此我们定义一个简单的文本生成方法 `generate_text_simple`:\n\n``` python\ndef generate_text_simple(model, idx, max_new_tokens, context_size):\n    return idx\n```\n\n这个方法就是最终我们需要输出新内容所需要的，我们对这个方法进行解释：\n  - `model` : 用来生成文本的大规模语言模型。这个模型经过训练，可以根据给定的输入（上下文）预测并生成接下来最有可能出现的文本。\n  - `idx` : 这通常指代的是输入文本序列中每个词或标记（token）在词汇表中的索引位置。在处理过程中，文本首先会被分词器转换成一系列的标记，然后每个标记会根据词汇表映射为一个索引值，用于模型的计算。\n  - `max_new_tokens` : 这是设定的一个上限，表示模型在生成新文本时最多可以输出的新标记数量。它有助于控制生成文本的长度，避免生成过长的内容。\n  - `context_size` : 模型在做预测时所参考的历史信息长度，即每次提供给模型的输入序列的长度。较大的上下文大小可以让模型记住更多过去的信息，从而可能生成更加连贯和有意义的文本。\n\n我们试着简单的实现这个方法：\n- 首先定义 `model` 参数的类型，名字为 `GPTModel`, 因为我们不知道如何实现它，姑且暂时将输入作为输出返回:\n\n``` python\nimport torch\nimport torch.nn as nn\nprint(\"torch version:\",torch.__version__)\n\nclass GPTModel(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, in_idx):\n        return in_idx\n\n```\n\n- 然后实现 `generate_text_simple` 方法:\n\n``` python\nimport random\ndef generate_text_simple(model, idx, max_new_tokens, context_size):\n\n    for _ in range(max_new_tokens):\n        # 获取上下文token\n        idx_cond = idx[-context_size:] \n        # 通过模型生成后续序列\n        logits = model(idx_cond) \n        # 随机选择一个\n        idx_next = random.choices(logits)[0] \n        #将下一个预测添加到序列中\n        idx.append(idx_next) \n\n    return idx\n```\n\n- 最后我们进行测试，并且有了输出，最大续写5个字，每次只处理上下文4个长度：\n\n``` python\n\nmodel = GPTModel()\nin_idx = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]\ngenerate_text_simple(model=model, idx=in_idx, max_new_tokens=5, context_size=4)\n\n```\n输出\n\n``` python\n\n[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 13, 16, 16, 16, 16]\n\n```\n\n但这个结果人类看不懂，并且是很随机的输出，所以它毫无意义。因此我们需要将这个模型进行完善，分析以上过程和代码，我们需要解决的问题有：\n- 将输入文本转为词汇表的索引位置\n- 实现模型预测，既完成模型训练\n- 选择最有可能或最优的预测，而不是随机输出\n- 使用正常的文本来测试，且将结果输出为人类能看懂的文本\n\n## 1. 数据准备与预处理\n\n### 1.1 简单分词器\n首先，要解决的是将文本转换为数值表示。为此我们需要一份词汇表，现在有很多公开的词汇表可以使用。但为了学习目的，我们从一份中文内容中自定义词汇表，当然这篇中文文章也是后续训练数据集。\n\n- 加载我们要处理的原始文本\n- 《每一滴汗水都是未来花朵的养分》是由GPT根据我的提示词生成的文章。\n\n```python\nwith open(\"the-road.txt\", \"r\", encoding=\"utf-8\") as f:\n    raw_text = f.read()\n\nprint(\"总字数：\", len(raw_text))\nprint(raw_text[:99])\n```\n输出\n```python\n总字数： 778\n每一滴汗水都是未来花朵的养分——在生命的旅途上，我们每个人都是自己故事中的主角。每一天，我们都面临着选择，每一个选择都是一条通往未知的道路。而在这无数的选择中，有一种选择是永恒不变的——那就是付出努\n['每', '一', '滴', '汗', '水', '都', '是', '未', '来', '花', '朵', '的', '养', '分', '—', '—', '在', '生', '命', '的', '旅', '途', '上', '，', '我', '们', '每', '个', '人', '都', '是', '自', '己', '故', '事', '中', '的', '主', '角', '。', '每', '一', '天', '，', '我', '们', '都', '面', '临', '着', '选', '择', '，', '每', '一', '个', '选', '择', '都', '是', '一', '条', '通', '往', '未', '知', '的', '道', '路', '。', '而', '在', '这', '无', '数', '的', '选', '择', '中', '，', '有', '一', '种', '选', '择', '是', '永', '恒', '不', '变', '的', '—', '—', '那', '就', '是', '付', '出', '努', '力', '。', '每', '一', '次', '的', '努', '力', '，', '无', '论', '大', '小', '，', '都', '在', '悄', '然', '改', '变', '着', '我', '们', '的', '命', '运', '轨', '迹', '。', '它', '们', '累', '积', '起', '来', '，', '成', '为', '我', '们', '成', '长', '道', '路', '上', '最', '宝', '贵', '的', '财', '富', '。', '每', '一', '次', '努', '力', '都', '让', '你', '感', '动', '。', '这', '不', '仅', '仅', '是', '对', '个', '人', '成', '就', '的', '一', '种', '赞', '美', '，', '更', '是', '对', '坚', '持', '和', '毅', '力', '的', '颂', '扬', '。', '当', '我', '们', '看', '到', '运', '动', '员', '在', '赛', '场', '上', '挥', '洒', '汗', '水', '，', '科', '学', '家', '在', '实', '验', '室', '里', '日', '夜', '钻', '研', '，', '或', '是', '普', '通', '劳', '动', '者', '在', '岗', '位', '上', '默', '默', '耕', '耘', '时', '，', '我', '们', '内', '心', '深', '处', '都', '会', '被', '触', '动', '。', '这', '种', '感', '动', '源', '于', '对', '人', '类', '潜', '能', '的', '敬', '佩', '，', '以', '及', '对', '那', '些', '为', '了', '梦', '想', '而', '不', '惜', '一', '切', '代', '价', '的', '人', '们', '的', '尊', '重', '。', '未', '来', '的', '精', '彩', '由', '此', '慢', '慢', '绽', '放', '。', '就', '像', '一', '朵', '花', '从', '种', '子', '开', '始', '，', '在', '土', '壤', '中', '吸', '收', '养', '分', '，', '经', '过', '长', '时', '间', '的', '孕', '育', '，', '最', '终', '破', '土', '而', '出', '，', '展', '现', '出', '它', '的', '美', '丽', '一', '样', '。', '我', '们', '的', '未', '来', '也', '是', '这', '样', '，', '通', '过', '不', '断', '的', '努', '力', '，', '我', '们', '将', '积', '累', '起', '足', '够', '的', '力', '量', '，', '去', '迎', '接', '即', '将', '到', '来', '的', '挑', '战', '与', '机', '遇', '。', '每', '一', '次', '小', '小', '的', '进', '步', '，', '每', '一', '点', '微', '不', '足', '道', '的', '成', '功', '，', '都', '是', '通', '向', '更', '大', '辉', '煌', '的', '基', '石', '。', '在', '这', '个', '过', '程', '中', '，', '我', '们', '需', '要', '明', '白', '，', '成', '功', '并', '非', '一', '蹴', '而', '就', '。', '它', '是', '由', '无', '数', '次', '失', '败', '、', '反', '思', '和', '再', '尝', '试', '构', '成', '的', '。', '因', '此', '，', '当', '我', '们', '在', '追', '求', '梦', '想', '的', '路', '上', '遇', '到', '困', '难', '时', '，', '不', '要', '轻', '易', '放', '弃', '。', '相', '反', '，', '应', '该', '把', '每', '一', '次', '挫', '折', '看', '作', '是', '学', '习', '的', '机', '会', '，', '从', '中', '吸', '取', '教', '训', '，', '并', '且', '更', '加', '坚', '定', '地', '朝', '着', '目', '标', '前', '进', '。', '同', '时', '，', '我', '们', '也', '应', '当', '珍', '惜', '身', '边', '那', '些', '支', '持', '和', '鼓', '励', '我', '们', '的', '人', '。', '他', '们', '可', '能', '是', '家', '人', '、', '朋', '友', '、', '老', '师', '或', '者', '是', '陌', '生', '人', '。', '正', '是', '有', '了', '他', '们', '的', '陪', '伴', '和', '支', '持', '，', '我', '们', '才', '能', '更', '有', '勇', '气', '面', '对', '生', '活', '中', '的', '风', '风', '雨', '雨', '。', '所', '以', '，', '让', '我', '们', '心', '怀', '感', '恩', '之', '心', '，', '用', '实', '际', '行', '动', '回', '报', '那', '些', '帮', '助', '过', '我', '们', '的', '人', '。', '总', '之', '，', '“', '每', '一', '次', '努', '力', '都', '让', '你', '感', '动', ',', '未', '来', '的', '精', '彩', '由', '此', '慢', '慢', '绽', '放', '”', '这', '句', '话', '不', '仅', '仅', '是', '一', '句', '简', '单', '的', '口', '号', '，', '它', '蕴', '含', '着', '深', '刻', '的', '人', '生', '哲', '理', '。', '它', '提', '醒', '我', '们', '要', '珍', '惜', '每', '一', '个', '当', '下', '，', '用', '心', '去', '感', '受', '生', '命', '中', '的', '点', '滴', '美', '好', '；', '同', '时', '也', '激', '励', '我', '们', '在', '面', '对', '困', '难', '时', '不', '退', '缩', '，', '勇', '往', '直', '前', '。', '因', '为', '只', '有', '经', '历', '了', '风', '雨', '洗', '礼', '后', '的', '彩', '虹', '，', '才', '会', '显', '得', '格', '外', '绚', '丽', '多', '彩', '。']\n```\n\n- 然后我们需要将这篇文章 `token` 化，考虑一些常见的中文标点符号\n\n```python\nimport re\npattern = r'([\\u4e00-\\u9fff，。_！？、；：“”‘’()（）——])' # 匹配常见的中文标点符号\nresult = [token for token in re.split(pattern, raw_text) if token.strip()] # 分割并移除空字符串\nprint(result)\nprint(len(result))\n```\n输出\n```python\n['每', '一', '滴', '汗', '水', '都', '是', '未', '来', '花', '朵', '的', '养', '分', '—', '—', '在', '生', '命', '的', '旅', '途', '上', '，', '我', '们', '每', '个', '人', '都', '是', '自', '己', '故', '事', '中', '的', '主', '角', '。', '每', '一', '天', '，', '我', '们', '都', '面', '临', '着', '选', '择', '，', '每', '一', '个', '选', '择', '都', '是', '一', '条', '通', '往', '未', '知', '的', '道', '路', '。', '而', '在', '这', '无', '数', '的', '选', '择', '中', '，', '有', '一', '种', '选', '择', '是', '永', '恒', '不', '变', '的', '—', '—', '那', '就', '是', '付', '出', '努', '力', '。', '每', '一', '次', '的', '努', '力', '，', '无', '论', '大', '小', '，', '都', '在', '悄', '然', '改', '变', '着', '我', '们', '的', '命', '运', '轨', '迹', '。', '它', '们', '累', '积', '起', '来', '，', '成', '为', '我', '们', '成', '长', '道', '路', '上', '最', '宝', '贵', '的', '财', '富', '。', '每', '一', '次', '努', '力', '都', '让', '你', '感', '动', '。', '这', '不', '仅', '仅', '是', '对', '个', '人', '成', '就', '的', '一', '种', '赞', '美', '，', '更', '是', '对', '坚', '持', '和', '毅', '力', '的', '颂', '扬', '。', '当', '我', '们', '看', '到', '运', '动', '员', '在', '赛', '场', '上', '挥', '洒', '汗', '水', '，', '科', '学', '家', '在', '实', '验', '室', '里', '日', '夜', '钻', '研', '，', '或', '是', '普', '通', '劳', '动', '者', '在', '岗', '位', '上', '默', '默', '耕', '耘', '时', '，', '我', '们', '内', '心', '深', '处', '都', '会', '被', '触', '动', '。', '这', '种', '感', '动', '源', '于', '对', '人', '类', '潜', '能', '的', '敬', '佩', '，', '以', '及', '对', '那', '些', '为', '了', '梦', '想', '而', '不', '惜', '一', '切', '代', '价', '的', '人', '们', '的', '尊', '重', '。', '未', '来', '的', '精', '彩', '由', '此', '慢', '慢', '绽', '放', '。', '就', '像', '一', '朵', '花', '从', '种', '子', '开', '始', '，', '在', '土', '壤', '中', '吸', '收', '养', '分', '，', '经', '过', '长', '时', '间', '的', '孕', '育', '，', '最', '终', '破', '土', '而', '出', '，', '展', '现', '出', '它', '的', '美', '丽', '一', '样', '。', '我', '们', '的', '未', '来', '也', '是', '这', '样', '，', '通', '过', '不', '断', '的', '努', '力', '，', '我', '们', '将', '积', '累', '起', '足', '够', '的', '力', '量', '，', '去', '迎', '接', '即', '将', '到', '来', '的', '挑', '战', '与', '机', '遇', '。', '每', '一', '次', '小', '小', '的', '进', '步', '，', '每', '一', '点', '微', '不', '足', '道', '的', '成', '功', '，', '都', '是', '通', '向', '更', '大', '辉', '煌', '的', '基', '石', '。', '在', '这', '个', '过', '程', '中', '，', '我', '们', '需', '要', '明', '白', '，', '成', '功', '并', '非', '一', '蹴', '而', '就', '。', '它', '是', '由', '无', '数', '次', '失', '败', '、', '反', '思', '和', '再', '尝', '试', '构', '成', '的', '。', '因', '此', '，', '当', '我', '们', '在', '追', '求', '梦', '想', '的', '路', '上', '遇', '到', '困', '难', '时', '，', '不', '要', '轻', '易', '放', '弃', '。', '相', '反', '，', '应', '该', '把', '每', '一', '次', '挫', '折', '看', '作', '是', '学', '习', '的', '机', '会', '，', '从', '中', '吸', '取', '教', '训', '，', '并', '且', '更', '加', '坚', '定', '地', '朝', '着', '目', '标', '前', '进', '。', '同', '时', '，', '我', '们', '也', '应', '当', '珍', '惜', '身', '边', '那', '些', '支', '持', '和', '鼓', '励', '我', '们', '的', '人', '。', '他', '们', '可', '能', '是', '家', '人', '、', '朋', '友', '、', '老', '师', '或', '者', '是', '陌', '生', '人', '。', '正', '是', '有', '了', '他', '们', '的', '陪', '伴', '和', '支', '持', '，', '我', '们', '才', '能', '更', '有', '勇', '气', '面', '对', '生', '活', '中', '的', '风', '风', '雨', '雨', '。', '所', '以', '，', '让', '我', '们', '心', '怀', '感', '恩', '之', '心', '，', '用', '实', '际', '行', '动', '回', '报', '那', '些', '帮', '助', '过', '我', '们', '的', '人', '。', '总', '之', '，', '“', '每', '一', '次', '努', '力', '都', '让', '你', '感', '动', ',', '未', '来', '的', '精', '彩', '由', '此', '慢', '慢', '绽', '放', '”', '这', '句', '话', '不', '仅', '仅', '是', '一', '句', '简', '单', '的', '口', '号', '，', '它', '蕴', '含', '着', '深', '刻', '的', '人', '生', '哲', '理', '。', '它', '提', '醒', '我', '们', '要', '珍', '惜', '每', '一', '个', '当', '下', '，', '用', '心', '去', '感', '受', '生', '命', '中', '的', '点', '滴', '美', '好', '；', '同', '时', '也', '激', '励', '我', '们', '在', '面', '对', '困', '难', '时', '不', '退', '缩', '，', '勇', '往', '直', '前', '。', '因', '为', '只', '有', '经', '历', '了', '风', '雨', '洗', '礼', '后', '的', '彩', '虹', '，', '才', '会', '显', '得', '格', '外', '绚', '丽', '多', '彩', '。']\n768\n```\n\n- 从这 `768` 个 `token` 中，我们构建一个包含所有唯一 `token` 的词汇表：\n\n```python\nall_tokens = sorted(set(result))\nvocab = {token : integer for integer,token  in enumerate(all_tokens)}\nvocab_size = len(vocab)\nprint(vocab_size)\n```\n输出\n```python\n321\n```\n\n打印前后5个元素看一下效果：\n\n```python\nfor i, item in enumerate(vocab.items()):\n    if i \u003c= 5 or i \u003e= (vocab_size-5):\n        print(item)\n    elif i == 6 or i == (vocab_size-5):\n        print('...')\n```\n输出\n```python\n(',', 0)\n('—', 1)\n('“', 2)\n('”', 3)\n('、', 4)\n('。', 5)\n...\n('验', 316)\n('默', 317)\n('鼓', 318)\n('，', 319)\n('；', 320)\n```\n\n- 接下来我们要用这个词汇表将文本进行 `token` 化，可以理解成进行 `编码`, 当然后续也需要将这些 `token` ID 转化回文本形式，即进行 `解码`，所以我们将实现这两个方法，并放到一个 `SimpleTokenizer` 类中。其中，`encode` 实现将文本转换为  `token` ID, `decode` 实现将 `token` ID 重新转换为文本：\n\n```python\nclass SimpleTokenizer:\n    def __init__(self, vocab):\n        self.str_to_int = vocab\n        self.int_to_str = {i:s for s,i in vocab.items()}\n\n    def encode(self, text):\n        pattern = r'([\\u4e00-\\u9fff，。_！？、；：“”‘’()（）——])' # 匹配常见的中文标点符号\n        preprocessed = [item.strip() for item in re.split(pattern, text) if item.strip()] # 分割并移除空字符串\n        ids = [self.str_to_int[s] for s in preprocessed]\n        return ids\n\n    def decode(self, ids):\n        text = \"\".join([self.int_to_str[i] for i in ids])\n        return text\n```\n\n我们可以使用 `SimpleTokenizer` 将文本编码为数值，然后可以将数值嵌入作为LLM的输入：\n\n```python\ntokenizer = SimpleTokenizer(vocab=vocab)\n\nids = tokenizer.encode(\"每一次努力都让你感动\")\nprint(ids)\n```\n输出\n```python\n[199, 6, 194, 55, 50, 298, 264, 38, 142, 53]\n```\n\n我们还可以将这些数值解码回文本：\n\n```python\ntokenizer.decode(ids)\n```\n输出\n```python\n'每一次努力都让你感动'\n```\n\n- 当然这个分词器还不完整，例如对于未知字没有特殊的处理，会导致代码运行错误，为此，我们可以添加一些特殊的标记：\n    - `\"\u003c|unk|\u003e\"` 表示未知单词\n    - `\"\u003c|endoftext|\u003e\"` 表示文本的结尾\n\n```python\nall_tokens = sorted(set(result))\nall_tokens.extend([\"\u003c|endoftext|\u003e\", \"\u003c|unk|\u003e\"])\nvocab = {token:integer for integer,token in enumerate(all_tokens)}\nprint(len(vocab.items()))\nfor i, item in enumerate(list(vocab.items())[-5:]):\n    print(item)\n```\n输出\n```python\n323\n('鼓', 318)\n('，', 319)\n('；', 320)\n('\u003c|endoftext|\u003e', 321)\n('\u003c|unk|\u003e', 322)\n```\n\n同时需要修改 `SimpleTokenizer`，以便知道何时以及如何使用新的`\u003cunk\u003e` 标记\n\n```python\nclass SimpleTokenizer:\n    def __init__(self, vocab):\n        self.str_to_int = vocab\n        self.int_to_str = {i:s for s,i in vocab.items()}\n\n    def encode(self, text):\n        pattern = r'([\\u4e00-\\u9fff，。_！？、；：“”‘’()（）——])' # 匹配常见的中文标点符号\n        preprocessed = [item.strip() for item in re.split(pattern, text) if item.strip()] # 分割并移除空字符串\n        preprocessed = [item if item in self.str_to_int else \"\u003c|unk|\u003e\" for item in preprocessed]\n        ids = [self.str_to_int[s] for s in preprocessed]\n        return ids\n\n    def decode(self, ids):\n        text = \"\".join([self.int_to_str[i] for i in ids])\n        return text\n```\n\n- **这个分词器有没有什么问题？** \n  - **当然，因为大部分情况下需要保留词语或短语，而不只仅仅单个字，它不会理解语义或语法结构，只是机械式的切割** \n  - **这里我们暂时以这个为主，后续来进行完善，假设我们分词器已完成，来完成剩余部分。**  \n\n\n让我们尝试使用修改后的标记器来标记文本：\n\n```python\ntokenizer = SimpleTokenizer(vocab=vocab)\n\nids = tokenizer.encode(\"每一次努力都让你感动啊\")\nprint(ids)\n```\n输出\n\n```python\n[199, 6, 194, 55, 50, 298, 264, 38, 142, 53, 322]\n```\n\n```python\ntokenizer.decode(ids)\n```\n输出\n```python\n'每一次努力都让你感动\u003c|unk|\u003e'\n```\n\n对整个原始文本进行编码：\n\n```python\nids = tokenizer.encode(raw_text + \"\u003c|endoftext|\u003e\")\nprint(ids)\n```\n输出\n\n```python\n[199, 6, 210, 204, 201, 298, 176, 184, 188, 256, 185, 223, 41, 45, 1, 1, 87, 219, 80, 223, 170, 293, 7, 319, 145, 32, 199, 12, 25, 298, 176, 255, 121, 165, 22, 13, 223, 16, 262, 5, 199, 6, 99, 319, 145, 32, 298, 313, 14, 228, 292, 154, 319, 199, 6, 12, 292, 154, 298, 176, 6, 187, 294, 130, 184, 229, 223, 296, 277, 5, 250, 87, 287, 171, 168, 223, 292, 154, 13, 319, 181, 6, 234, 292, 154, 176, 202, 137, 9, 68, 223, 1, 1, 297, 118, 176, 29, 44, 55, 50, 5, 199, 6, 194, 223, 55, 50, 319, 171, 266, 98, 116, 319, 298, 87, 139, 214, 163, 68, 228, 145, 32, 223, 80, 286, 280, 289, 5, 106, 32, 241, 236, 275, 188, 319, 144, 15, 145, 32, 144, 304, 296, 277, 7, 180, 108, 272, 223, 270, 112, 5, 199, 6, 194, 55, 50, 298, 264, 38, 142, 53, 5, 287, 9, 26, 26, 176, 113, 12, 25, 144, 118, 223, 6, 234, 274, 247, 319, 179, 176, 113, 90, 155, 81, 198, 50, 223, 314, 150, 5, 128, 145, 32, 227, 47, 286, 53, 79, 87, 273, 89, 7, 157, 205, 204, 201, 319, 235, 105, 111, 87, 109, 316, 110, 300, 172, 96, 303, 231, 319, 146, 176, 178, 294, 57, 53, 249, 87, 120, 36, 7, 317, 317, 251, 252, 173, 319, 145, 32, 42, 133, 208, 93, 298, 34, 260, 263, 53, 5, 287, 234, 142, 53, 209, 23, 113, 25, 239, 211, 254, 223, 167, 39, 319, 31, 63, 113, 297, 24, 15, 21, 193, 141, 250, 9, 140, 6, 46, 30, 33, 223, 25, 32, 223, 115, 301, 5, 184, 188, 223, 240, 129, 221, 196, 143, 143, 245, 164, 5, 118, 40, 6, 185, 256, 27, 234, 103, 126, 102, 319, 87, 86, 92, 13, 78, 162, 41, 45, 319, 243, 284, 304, 173, 305, 223, 104, 253, 319, 180, 242, 232, 86, 250, 44, 319, 119, 216, 44, 106, 223, 247, 17, 6, 191, 5, 145, 32, 223, 184, 188, 19, 176, 287, 191, 319, 294, 284, 9, 169, 223, 55, 50, 319, 145, 32, 114, 236, 241, 275, 276, 97, 223, 50, 302, 319, 62, 285, 159, 60, 114, 47, 188, 223, 156, 147, 10, 186, 295, 5, 199, 6, 194, 116, 116, 223, 288, 197, 319, 199, 6, 213, 132, 9, 276, 296, 223, 144, 51, 319, 298, 176, 294, 76, 179, 98, 282, 215, 223, 91, 230, 5, 87, 287, 12, 284, 237, 13, 319, 145, 32, 311, 261, 174, 222, 319, 144, 51, 124, 312, 6, 278, 250, 118, 5, 106, 176, 221, 171, 168, 194, 100, 271, 4, 65, 135, 81, 43, 117, 267, 189, 144, 223, 5, 84, 196, 319, 128, 145, 32, 87, 290, 203, 193, 141, 223, 277, 7, 295, 47, 85, 309, 173, 319, 9, 261, 281, 175, 164, 127, 5, 226, 65, 319, 125, 269, 151, 199, 6, 194, 158, 152, 227, 37, 176, 105, 20, 223, 186, 34, 319, 27, 13, 78, 66, 166, 265, 319, 124, 11, 179, 52, 90, 107, 88, 183, 228, 224, 190, 49, 288, 5, 74, 173, 319, 145, 32, 19, 125, 128, 217, 140, 279, 283, 297, 24, 161, 155, 81, 318, 56, 145, 32, 223, 25, 5, 28, 32, 72, 254, 176, 111, 25, 4, 182, 64, 4, 248, 122, 146, 249, 176, 307, 219, 25, 5, 195, 176, 181, 21, 28, 32, 223, 308, 35, 81, 161, 155, 319, 145, 32, 149, 254, 179, 181, 58, 200, 313, 113, 219, 207, 13, 223, 315, 315, 310, 310, 5, 148, 31, 319, 264, 145, 32, 133, 134, 142, 138, 18, 133, 319, 220, 109, 306, 259, 53, 83, 153, 297, 24, 123, 54, 284, 145, 32, 223, 25, 5, 136, 18, 319, 2, 199, 6, 194, 55, 50, 298, 264, 38, 142, 53, 0, 184, 188, 223, 240, 129, 221, 196, 143, 143, 245, 164, 3, 287, 70, 268, 9, 26, 26, 176, 6, 70, 238, 59, 223, 69, 73, 319, 106, 257, 77, 228, 208, 48, 223, 25, 219, 82, 218, 5, 106, 160, 299, 145, 32, 261, 217, 140, 199, 6, 12, 128, 8, 319, 220, 133, 62, 142, 67, 219, 80, 13, 223, 213, 210, 247, 101, 320, 74, 173, 19, 212, 56, 145, 32, 87, 313, 113, 85, 309, 173, 9, 291, 246, 319, 58, 130, 225, 49, 5, 84, 15, 71, 181, 243, 61, 21, 315, 310, 206, 233, 75, 223, 129, 258, 319, 149, 34, 177, 131, 192, 94, 244, 17, 95, 129, 5, 321]\n```\n\n\n```python\ntokenizer.decode(ids)\n```\n输出\n```python\n'每一滴汗水都是未来花朵的养分——在生命的旅途上，我们每个人都是自己故事中的主角。每一天，我们都面临着选择，每一个选择都是一条通往未知的道路。而在这无数的选择中，有一种选择是永恒不变的——那就是付出努力。每一次的努力，无论大小，都在悄然改变着我们的命运轨迹。它们累积起来，成为我们成长道路上最宝贵的财富。每一次努力都让你感动。这不仅仅是对个人成就的一种赞美，更是对坚持和毅力的颂扬。当我们看到运动员在赛场上挥洒汗水，科学家在实验室里日夜钻研，或是普通劳动者在岗位上默默耕耘时，我们内心深处都会被触动。这种感动源于对人类潜能的敬佩，以及对那些为了梦想而不惜一切代价的人们的尊重。未来的精彩由此慢慢绽放。就像一朵花从种子开始，在土壤中吸收养分，经过长时间的孕育，最终破土而出，展现出它的美丽一样。我们的未来也是这样，通过不断的努力，我们将积累起足够的力量，去迎接即将到来的挑战与机遇。每一次小小的进步，每一点微不足道的成功，都是通向更大辉煌的基石。在这个过程中，我们需要明白，成功并非一蹴而就。它是由无数次失败、反思和再尝试构成的。因此，当我们在追求梦想的路上遇到困难时，不要轻易放弃。相反，应该把每一次挫折看作是学习的机会，从中吸取教训，并且更加坚定地朝着目标前进。同时，我们也应当珍惜身边那些支持和鼓励我们的人。他们可能是家人、朋友、老师或者是陌生人。正是有了他们的陪伴和支持，我们才能更有勇气面对生活中的风风雨雨。所以，让我们心怀感恩之心，用实际行动回报那些帮助过我们的人。总之，“每一次努力都让你感动,未来的精彩由此慢慢绽放”这句话不仅仅是一句简单的口号，它蕴含着深刻的人生哲理。它提醒我们要珍惜每一个当下，用心去感受生命中的点滴美好；同时也激励我们在面对困难时不退缩，勇往直前。因为只有经历了风雨洗礼后的彩虹，才会显得格外绚丽多彩。\u003c|endoftext|\u003e'\n```\n\n### 1.2 处理文本数据\n最后，为了将该文本变成我们的第一个训练数据集，我们需要构建对它进行处理，以实现对模型的训练：\n\n- 第一步，构建输入输出组合，所谓输入输出组合就是根据前面的所有字来预测下一个字\n  为了说明这个问题，我们使用原始文档进行说明\n\n```python\nimport re\npreprocessed = [token for token in re.split(pattern, raw_text) if token.strip()] #将原始文档进行分割去除空字符串\n\nfor i in range(1, len(preprocessed)):\n    input = preprocessed[:i]\n    output = preprocessed[i]\n\n    print(\"\".join(input), \"——\u003e\", output)\n\n```\n输出\n\n```python\n每 ——\u003e 一\n每一 ——\u003e 滴\n每一滴 ——\u003e 汗\n每一滴汗 ——\u003e 水\n每一滴汗水 ——\u003e 都\n每一滴汗水都 ——\u003e 是\n每一滴汗水都是 ——\u003e 未\n每一滴汗水都是未 ——\u003e 来\n每一滴汗水都是未来 ——\u003e 花\n每一滴汗水都是未来花 ——\u003e 朵\n每一滴汗水都是未来花朵 ——\u003e 的\n每一滴汗水都是未来花朵的 ——\u003e 养\n每一滴汗水都是未来花朵的养 ——\u003e 分\n每一滴汗水都是未来花朵的养分 ——\u003e —\n每一滴汗水都是未来花朵的养分— ——\u003e —\n每一滴汗水都是未来花朵的养分—— ——\u003e 在\n每一滴汗水都是未来花朵的养分——在 ——\u003e 生\n每一滴汗水都是未来花朵的养分——在生 ——\u003e 命\n每一滴汗水都是未来花朵的养分——在生命 ——\u003e 的\n每一滴汗水都是未来花朵的养分——在生命的 ——\u003e 旅\n每一滴汗水都是未来花朵的养分——在生命的旅 ——\u003e 途\n每一滴汗水都是未来花朵的养分——在生命的旅途 ——\u003e 上\n每一滴汗水都是未来花朵的养分——在生命的旅途上 ——\u003e ，\n每一滴汗水都是未来花朵的养分——在生命的旅途上， ——\u003e 我\n每一滴汗水都是未来花朵的养分——在生命的旅途上，我 ——\u003e 们\n...\n每一滴汗水都是未来花朵的养分——在生命的旅途上，我们每个人都是自己故事中的主角。每一天，我们都面临着选择，每一个选择都是一条通往未知的道路。而在这无数的选择中，有一种选择是永恒不变的——那就是付出努力。每一次的努力，无论大小，都在悄然改变着我们的命运轨迹。它们累积起来，成为我们成长道路上最宝贵的财富。每一次努力都让你感动。这不仅仅是对个人成就的一种赞美，更是对坚持和毅力的颂扬。当我们看到运动员在赛场上挥洒汗水，科学家在实验室里日夜钻研，或是普通劳动者在岗位上默默耕耘时，我们内心深处都会被触动。这种感动源于对人类潜能的敬佩，以及对那些为了梦想而不惜一切代价的人们的尊重。未来的精彩由此慢慢绽放。就像一朵花从种子开始，在土壤中吸收养分，经过长时间的孕育，最终破土而出，展现出它的美丽一样。我们的未来也是这样，通过不断的努力，我们将积累起足够的力量，去迎接即将到来的挑战与机遇。每一次小小的进步，每一点微不足道的成功，都是通向更大辉煌的基石。在这个过程中，我们需要明白，成功并非一蹴而就。它是由无数次失败、反思和再尝试构成的。因此，当我们在追求梦想的路上遇到困难时，不要轻易放弃。相反，应该把每一次挫折看作是学习的机会，从中吸取教训，并且更加坚定地朝着目标前进。同时，我们也应当珍惜身边那些支持和鼓励我们的人。他们可能是家人、朋友、老师或者是陌生人。正是有了他们的陪伴和支持，我们才能更有勇气面对生活中的风风雨雨。所以，让我们心怀感恩之心，用实际行动回报那些帮助过我们的人。总之，“每一次努力都让你感动,未来的精彩由此慢慢绽放”这句话不仅仅是一句简单的口号，它蕴含着深刻的人生哲理。它提醒我们要珍惜每一个当下，用心去感受生命中的点滴美好；同时也激励我们在面对困难时不退缩，勇往直前。因为只有经历了风雨洗礼后的彩虹，才会显得格外绚 ——\u003e 丽\n每一滴汗水都是未来花朵的养分——在生命的旅途上，我们每个人都是自己故事中的主角。每一天，我们都面临着选择，每一个选择都是一条通往未知的道路。而在这无数的选择中，有一种选择是永恒不变的——那就是付出努力。每一次的努力，无论大小，都在悄然改变着我们的命运轨迹。它们累积起来，成为我们成长道路上最宝贵的财富。每一次努力都让你感动。这不仅仅是对个人成就的一种赞美，更是对坚持和毅力的颂扬。当我们看到运动员在赛场上挥洒汗水，科学家在实验室里日夜钻研，或是普通劳动者在岗位上默默耕耘时，我们内心深处都会被触动。这种感动源于对人类潜能的敬佩，以及对那些为了梦想而不惜一切代价的人们的尊重。未来的精彩由此慢慢绽放。就像一朵花从种子开始，在土壤中吸收养分，经过长时间的孕育，最终破土而出，展现出它的美丽一样。我们的未来也是这样，通过不断的努力，我们将积累起足够的力量，去迎接即将到来的挑战与机遇。每一次小小的进步，每一点微不足道的成功，都是通向更大辉煌的基石。在这个过程中，我们需要明白，成功并非一蹴而就。它是由无数次失败、反思和再尝试构成的。因此，当我们在追求梦想的路上遇到困难时，不要轻易放弃。相反，应该把每一次挫折看作是学习的机会，从中吸取教训，并且更加坚定地朝着目标前进。同时，我们也应当珍惜身边那些支持和鼓励我们的人。他们可能是家人、朋友、老师或者是陌生人。正是有了他们的陪伴和支持，我们才能更有勇气面对生活中的风风雨雨。所以，让我们心怀感恩之心，用实际行动回报那些帮助过我们的人。总之，“每一次努力都让你感动,未来的精彩由此慢慢绽放”这句话不仅仅是一句简单的口号，它蕴含着深刻的人生哲理。它提醒我们要珍惜每一个当下，用心去感受生命中的点滴美好；同时也激励我们在面对困难时不退缩，勇往直前。因为只有经历了风雨洗礼后的彩虹，才会显得格外绚丽 ——\u003e 多\n每一滴汗水都是未来花朵的养分——在生命的旅途上，我们每个人都是自己故事中的主角。每一天，我们都面临着选择，每一个选择都是一条通往未知的道路。而在这无数的选择中，有一种选择是永恒不变的——那就是付出努力。每一次的努力，无论大小，都在悄然改变着我们的命运轨迹。它们累积起来，成为我们成长道路上最宝贵的财富。每一次努力都让你感动。这不仅仅是对个人成就的一种赞美，更是对坚持和毅力的颂扬。当我们看到运动员在赛场上挥洒汗水，科学家在实验室里日夜钻研，或是普通劳动者在岗位上默默耕耘时，我们内心深处都会被触动。这种感动源于对人类潜能的敬佩，以及对那些为了梦想而不惜一切代价的人们的尊重。未来的精彩由此慢慢绽放。就像一朵花从种子开始，在土壤中吸收养分，经过长时间的孕育，最终破土而出，展现出它的美丽一样。我们的未来也是这样，通过不断的努力，我们将积累起足够的力量，去迎接即将到来的挑战与机遇。每一次小小的进步，每一点微不足道的成功，都是通向更大辉煌的基石。在这个过程中，我们需要明白，成功并非一蹴而就。它是由无数次失败、反思和再尝试构成的。因此，当我们在追求梦想的路上遇到困难时，不要轻易放弃。相反，应该把每一次挫折看作是学习的机会，从中吸取教训，并且更加坚定地朝着目标前进。同时，我们也应当珍惜身边那些支持和鼓励我们的人。他们可能是家人、朋友、老师或者是陌生人。正是有了他们的陪伴和支持，我们才能更有勇气面对生活中的风风雨雨。所以，让我们心怀感恩之心，用实际行动回报那些帮助过我们的人。总之，“每一次努力都让你感动,未来的精彩由此慢慢绽放”这句话不仅仅是一句简单的口号，它蕴含着深刻的人生哲理。它提醒我们要珍惜每一个当下，用心去感受生命中的点滴美好；同时也激励我们在面对困难时不退缩，勇往直前。因为只有经历了风雨洗礼后的彩虹，才会显得格外绚丽多 ——\u003e 彩\n每一滴汗水都是未来花朵的养分——在生命的旅途上，我们每个人都是自己故事中的主角。每一天，我们都面临着选择，每一个选择都是一条通往未知的道路。而在这无数的选择中，有一种选择是永恒不变的——那就是付出努力。每一次的努力，无论大小，都在悄然改变着我们的命运轨迹。它们累积起来，成为我们成长道路上最宝贵的财富。每一次努力都让你感动。这不仅仅是对个人成就的一种赞美，更是对坚持和毅力的颂扬。当我们看到运动员在赛场上挥洒汗水，科学家在实验室里日夜钻研，或是普通劳动者在岗位上默默耕耘时，我们内心深处都会被触动。这种感动源于对人类潜能的敬佩，以及对那些为了梦想而不惜一切代价的人们的尊重。未来的精彩由此慢慢绽放。就像一朵花从种子开始，在土壤中吸收养分，经过长时间的孕育，最终破土而出，展现出它的美丽一样。我们的未来也是这样，通过不断的努力，我们将积累起足够的力量，去迎接即将到来的挑战与机遇。每一次小小的进步，每一点微不足道的成功，都是通向更大辉煌的基石。在这个过程中，我们需要明白，成功并非一蹴而就。它是由无数次失败、反思和再尝试构成的。因此，当我们在追求梦想的路上遇到困难时，不要轻易放弃。相反，应该把每一次挫折看作是学习的机会，从中吸取教训，并且更加坚定地朝着目标前进。同时，我们也应当珍惜身边那些支持和鼓励我们的人。他们可能是家人、朋友、老师或者是陌生人。正是有了他们的陪伴和支持，我们才能更有勇气面对生活中的风风雨雨。所以，让我们心怀感恩之心，用实际行动回报那些帮助过我们的人。总之，“每一次努力都让你感动,未来的精彩由此慢慢绽放”这句话不仅仅是一句简单的口号，它蕴含着深刻的人生哲理。它提醒我们要珍惜每一个当下，用心去感受生命中的点滴美好；同时也激励我们在面对困难时不退缩，勇往直前。因为只有经历了风雨洗礼后的彩虹，才会显得格外绚丽多彩 ——\u003e 。\n```\n\n实际上，输入和输出都是固定的长度，例如输入选择前面4个字，输出是输入向后移动1个字，这样有利于计算机处理，否则在运算时候对于不固定的长度处理复杂度较高，现在修改一下程序：\n\n```python\ncontext_size = 4\nfor i in range(0, len(preprocessed) - context_size): # -context_size 是因为最后一组不足以构建一个组合\n    # 输入取固定的长度\n    input = preprocessed[i:i+context_size]\n\n    # 输出是输入向后移动一个字的固定长度\n    output = preprocessed[i+1:i+context_size+1]\n\n    print(\"\".join(input), \"——\u003e\", \"\".join(output))\n\n```\n输出\n```python\n每一滴汗 ——\u003e 一滴汗水\n一滴汗水 ——\u003e 滴汗水都\n滴汗水都 ——\u003e 汗水都是\n汗水都是 ——\u003e 水都是未\n水都是未 ——\u003e 都是未来\n都是未来 ——\u003e 是未来花\n是未来花 ——\u003e 未来花朵\n未来花朵 ——\u003e 来花朵的\n来花朵的 ——\u003e 花朵的养\n花朵的养 ——\u003e 朵的养分\n朵的养分 ——\u003e 的养分—\n的养分— ——\u003e 养分——\n养分—— ——\u003e 分——在\n分——在 ——\u003e ——在生\n——在生 ——\u003e —在生命\n—在生命 ——\u003e 在生命的\n在生命的 ——\u003e 生命的旅\n生命的旅 ——\u003e 命的旅途\n命的旅途 ——\u003e 的旅途上\n的旅途上 ——\u003e 旅途上，\n旅途上， ——\u003e 途上，我\n途上，我 ——\u003e 上，我们\n上，我们 ——\u003e ，我们每\n，我们每 ——\u003e 我们每个\n我们每个 ——\u003e 们每个人\n...\n得格外绚 ——\u003e 格外绚丽\n格外绚丽 ——\u003e 外绚丽多\n外绚丽多 ——\u003e 绚丽多彩\n绚丽多彩 ——\u003e 丽多彩。\n```\n\n另外，我们还可以给这个数据加载方法设置一个移动幅度，比如每次移动两个字：\n\n```python\ncontext_size = 4\nstride = 2\nfor i in range(0, len(preprocessed) - context_size, stride): # -context_size 是因为最后一组不足以构建一个组合\n    # 输入取固定的长度\n    input = preprocessed[i : i + context_size]\n\n    # 输出是输入向后移动一个字的固定长度\n    output = preprocessed[i + 1 : i + context_size + 1]\n\n    print(\"\".join(input), \"——\u003e\", \"\".join(output))\n\n```\n输出\n```python\n每一滴汗 ——\u003e 一滴汗水\n滴汗水都 ——\u003e 汗水都是\n水都是未 ——\u003e 都是未来\n是未来花 ——\u003e 未来花朵\n来花朵的 ——\u003e 花朵的养\n朵的养分 ——\u003e 的养分—\n养分—— ——\u003e 分——在\n——在生 ——\u003e —在生命\n在生命的 ——\u003e 生命的旅\n命的旅途 ——\u003e 的旅途上\n旅途上， ——\u003e 途上，我\n上，我们 ——\u003e ，我们每\n我们每个 ——\u003e 们每个人\n每个人都 ——\u003e 个人都是\n人都是自 ——\u003e 都是自己\n是自己故 ——\u003e 自己故事\n己故事中 ——\u003e 故事中的\n事中的主 ——\u003e 中的主角\n的主角。 ——\u003e 主角。每\n角。每一 ——\u003e 。每一天\n每一天， ——\u003e 一天，我\n天，我们 ——\u003e ，我们都\n我们都面 ——\u003e 们都面临\n都面临着 ——\u003e 面临着选\n临着选择 ——\u003e 着选择，\n...\n，才会显 ——\u003e 才会显得\n会显得格 ——\u003e 显得格外\n得格外绚 ——\u003e 格外绚丽\n外绚丽多 ——\u003e 绚丽多彩\n```\n\n- 第二步，通过上一节的简单分词器进行 `token` 化\n```python\nimport re\ncontext_size = 4\nstride = 2\nfor i in range(0, len(preprocessed) - context_size, stride): # -context_size 是因为最后一组不足以构建一个组合\n    # 输入取固定的长度\n    input = preprocessed[i:i+context_size]\n\n    # 输出是输入向后移动一个字的固定长度\n    output = preprocessed[i+1:i+context_size+1]\n\n    print(tokenizer.encode(\"\".join(input)), \"----\u003e\", tokenizer.encode(\"\".join(output)))\n```\n输出\n```python\n[199, 6, 210, 204] ----\u003e [6, 210, 204, 201]\n[210, 204, 201, 298] ----\u003e [204, 201, 298, 176]\n[201, 298, 176, 184] ----\u003e [298, 176, 184, 188]\n[176, 184, 188, 256] ----\u003e [184, 188, 256, 185]\n[188, 256, 185, 223] ----\u003e [256, 185, 223, 41]\n[185, 223, 41, 45] ----\u003e [223, 41, 45, 1]\n[41, 45, 1, 1] ----\u003e [45, 1, 1, 87]\n[1, 1, 87, 219] ----\u003e [1, 87, 219, 80]\n[87, 219, 80, 223] ----\u003e [219, 80, 223, 170]\n[80, 223, 170, 293] ----\u003e [223, 170, 293, 7]\n[170, 293, 7, 319] ----\u003e [293, 7, 319, 145]\n[7, 319, 145, 32] ----\u003e [319, 145, 32, 199]\n[145, 32, 199, 12] ----\u003e [32, 199, 12, 25]\n[199, 12, 25, 298] ----\u003e [12, 25, 298, 176]\n[25, 298, 176, 255] ----\u003e [298, 176, 255, 121]\n[176, 255, 121, 165] ----\u003e [255, 121, 165, 22]\n[121, 165, 22, 13] ----\u003e [165, 22, 13, 223]\n[22, 13, 223, 16] ----\u003e [13, 223, 16, 262]\n[223, 16, 262, 5] ----\u003e [16, 262, 5, 199]\n[262, 5, 199, 6] ----\u003e [5, 199, 6, 99]\n[199, 6, 99, 319] ----\u003e [6, 99, 319, 145]\n[99, 319, 145, 32] ----\u003e [319, 145, 32, 298]\n[145, 32, 298, 313] ----\u003e [32, 298, 313, 14]\n[298, 313, 14, 228] ----\u003e [313, 14, 228, 292]\n[14, 228, 292, 154] ----\u003e [228, 292, 154, 319]\n...\n[319, 149, 34, 177] ----\u003e [149, 34, 177, 131]\n[34, 177, 131, 192] ----\u003e [177, 131, 192, 94]\n[131, 192, 94, 244] ----\u003e [192, 94, 244, 17]\n[94, 244, 17, 95] ----\u003e [244, 17, 95, 129]\n```\n\n- 最后，我们创建一个 `GPTDataset` 来实现数据加载，从输入文本中提取块\n```python\nimport torch\nfrom torch.utils.data import Dataset, DataLoader\nclass GPTDataset(Dataset):\n    def __init__(self, text, tokenizer, context_size, stride):\n        self.input_ids = []\n        self.output_ids = []\n\n        token_ids = tokenizer.encode(text)\n\n        for i in range(0, len(token_ids) - context_size, stride): # -context_size 是因为最后一组不足以构建一个组合\n            # 输入取固定的长度\n            input = token_ids[i:i+context_size]\n\n            # 输出是输入向后移动一个字的固定长度\n            output = token_ids[i+1:i+context_size+1]\n\n            self.input_ids.append(torch.tensor(input))\n            self.output_ids.append(torch.tensor(output))\n    \n    def __len__(self):\n        return len(self.input_ids)\n    \n    def __getitem__(self, index):\n        return self.input_ids[index], self.output_ids[index]\n\n```\n\n使用 `create_dataloader` 来创建一个数据加载器，注意这里有个参数 `batch_size`，指的是批量大小，对数据集进行批量并行计算可提高效率，上面我们输出的那些例子都是 `batch_size=1` 的情况，实际训练中，这个参数可进行调整以提升效率，使结果更加稳定。\n\n```python\ndef create_dataloader(txt, vocab, batch_size=4, max_length=16, \n                         stride=8, shuffle=True, drop_last=True,\n                         num_workers=0):\n\n    # Initialize the tokenizer\n    tokenizer = SimpleTokenizer(vocab)\n\n    # Create dataset\n    dataset = GPTDataset(txt, tokenizer, max_length, stride)\n\n    # Create dataloader\n    dataloader = DataLoader(\n        dataset,\n        batch_size=batch_size,\n        shuffle=shuffle,\n        drop_last=drop_last,\n        num_workers=num_workers\n    )\n\n    return dataloader\n```\n\n使用 `create_vocab` 方法可临时获得词汇表\n\n```python\ndef create_vocab(raw_text):\n    pattern = r'([\\u4e00-\\u9fff，。_！？、；：“”‘’()（）——])' # 匹配常见的中文标点符号\n    preprocessed = [token for token in re.split(pattern, raw_text) if token.strip()] # 分割并移除空字符串\n\n    all_tokens = sorted(set(preprocessed))\n    all_tokens.extend([\"\u003c|endoftext|\u003e\", \"\u003c|unk|\u003e\"])\n    vocab = {token:integer for integer,token in enumerate(all_tokens)}\n\n    return vocab\n```\n\n- 让我们针对上下文大小为 4 的 LLM 测试批量大小为 1 的数据加载器：\n\n```python\nwith open(\"the-road.txt\", \"r\", encoding=\"utf-8\") as f:\n    raw_text = f.read()\n\nvocab = create_vocab(raw_text)\ncontext_size = 4\nstride = 1\nbatch_size = 1\ndataloader = create_dataloader(\n    raw_text, vocab=vocab, batch_size=batch_size, max_length=context_size, stride=stride, shuffle=False\n)\n\ndata_iter = iter(dataloader)\nfirst_batch = next(data_iter)\nprint(first_batch)\n```\n输出\n\n```python\n[tensor([[199,   6, 210, 204]]), tensor([[  6, 210, 204, 201]])]\n```\n\n```python\nsecond_batch = next(data_iter)\nprint(second_batch)\n```\n输出\n\n```python\n[tensor([[  6, 210, 204, 201]]), tensor([[210, 204, 201, 298]])]\n```\n\n\n- 我们还可以创建分批输出\n- 请注意，我们在这里增加了步幅，这样批次之间就不会出现重叠，因为更多的重叠可能会导致过度拟合\n\n```python\ndataloader = create_dataloader(raw_text, vocab=vocab, batch_size=8, max_length=4, stride=4, shuffle=False)\n\ndata_iter = iter(dataloader)\ninputs, targets = next(data_iter)\nprint(\"输入:\\n\", inputs)\nprint(\"\\n输出:\\n\", targets)\n```\n\n```python\n输入:\n tensor([[199,   6, 210, 204],\n        [201, 298, 176, 184],\n        [188, 256, 185, 223],\n        [ 41,  45,   1,   1],\n        [ 87, 219,  80, 223],\n        [170, 293,   7, 319],\n        [145,  32, 199,  12],\n        [ 25, 298, 176, 255]])\n\n输出:\n tensor([[  6, 210, 204, 201],\n        [298, 176, 184, 188],\n        [256, 185, 223,  41],\n        [ 45,   1,   1,  87],\n        [219,  80, 223, 170],\n        [293,   7, 319, 145],\n        [ 32, 199,  12,  25],\n        [298, 176, 255, 121]])\n```\n\n好了，到此数据准备完成，下一步我们进入模型训练\n\n## 2. 模型架构与实现\n本节我们将实现一个具体的模型，首先需要了解类GPT模型架构，这样有利于理解和生成文本，这部分是所有内容当中最抽象最难理解的部分，但要实现大模型，这部分又是绕不开的一个内容，我尽可能以易懂的文字来解释。\n\n \u003cimg src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch04_compressed/02.webp\" height=\"350px\"\u003e\n\n \u003cimg src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch04_compressed/13.webp?1\" height=\"350px\"\u003e\n\n \u003cimg src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch04_compressed/15.webp\" height=\"350px\"\u003e\n\n上图展示GPT模型架构，它是基于原始Transformer架构的解码器部分，实现按顺序生成单词。将原始文本经 `token` 化，输入到GPT模型中，经嵌入层、Transformer块、输出层，最后预测出下一个单词\n- `Tokenized Text`: 它是上一节中将文本 `token` 化为数值表示 `toekn ID`\n- `Embedding layers`: 嵌入层，其本质是对参数矩阵的查找操作，通常会在模型训练期间进行更新，主要包括`Token embedding layer`、`Positional embedding layer`；\n    - `Token embedding layer`: 标记嵌入层，\n    - `Positional embedding layer`:  位置嵌入层\n- `Transformer block`: 包括 `层归一化` (`LayerNorm`) 、`多头注意力机制`(`Masked multi-head attention`)、`Dropout` (`Dropout`)、`跳跃连接`(`Shortcut connection`)、`前馈网络`(`Feed forward`)、`Dropout` (`Dropout`)\n    - `层归一化` (`LayerNorm`)：将神经网络层的激活集中在平均值 0 附近，并将其方差归一化为 1，稳定训练并更快地收敛到有效权重\n    - `多头注意力机制`(`Masked multi-head attention`)：使用注意力机制可捕获数据依赖关系，而自注意力机制关注输入的不同部分\n    - `前馈网络`(`Feed forward`)：通过一系列线性变换和非线性激活函数对输入数据进行处理，以捕捉输入数据中的特征\n    - `跳跃连接`(`Shortcut connection`)：为梯度在网络中流动创建了一条替代的较短路径,通过将一层的输出添加到后一层的输出来实现的，通常会跳过中间的一个或多个层，以缓解梯度消失问题\n    - `Dropout` (`Dropout`)\n- `Output layers`: 输出层是一个线性层\n\n通过架构，我们自定义需要实现的各部分，主要包括 `GPTModel`,`LayerNorm`,`MultiHeadAttention`,`FeedForward`,`TransformerBlock`等，我们将之前定义的 `GPTModel` 类拿过来，并定义`LayerNorm`、`MultiHeadAttention`、`FeedForward`、`TransformerBlock`，接下来主要介绍各部分实现。\n\n```python\nimport torch\nimport torch.nn as nn\nprint(\"torch version:\",torch.__version__)\n\nclass GPTModel(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, x):\n        return x\n    \nclass LayerNorm(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, x):\n        return x\n\nclass MultiHeadAttention(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, x):\n        return x\n\nclass FeedForward(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, x):\n        return x\n\nclass TransformerBlock(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, x):\n        return x\n```\n\n在实现各部分前，我们先定义一组通用的参数：\n\n\n```python\nGPT_CONFIG_85M = {\n    \"vocab_size\": 323,      # 词汇表大小\n    \"context_length\": 8,    # 上下文长度\n    \"emb_dim\": 768,          # 嵌入维度\n    \"n_heads\": 12,          # 注意力头数量\n    \"n_layers\": 12,         # 层数\n    \"drop_rate\": 0.1,       # Dropout 率\n    \"qkv_bias\": False       # Query-Key-Value bias\n}\n```\n\n- 我们使用短变量名以避免以后出现长代码行\n- `\"vocab_size\"` 表示词汇量为 323 个字\n- `\"context_length\"` 表示模型的最大输入标记数\n- `\"emb_dim\"` 是标记输入的嵌入大小，将每个输入标记转换为 768 维向量\n- `\"n_heads\"` 多头注意机制中的注意头数量\n- `\"n_layers\"` 是模型中的 `transformer` 块数量\n- `\"drop_rate\"` dropout 机制的强度； 0.1 表示在训练期间丢弃 10% 的隐藏单元以减轻过度拟合\n- `\"qkv_bias\"` 决定多头注意力机制中的 `Linear` 层在计算查询（Q）、键（K）和值（V）张量时是否应包含偏差向量；\n\n### 2.1 实现嵌入层 \n\n- 标记嵌入（`Token embedding layer` ）可以理解为通过输入序列的索引位置在整个词汇表中找到所有输出特征的参数矩阵的初始值，代码如下：\n\n```pyhton\nvocab_size = len(vocab)\noutput_dim = 16 # 假定16个特征维度\n\ntorch.manual_seed(123)\ntoken_embedding_layer = torch.nn.Embedding(vocab_size, output_dim) # 初始化一个嵌入层\n\ntoken_embeddings = token_embedding_layer(inputs) # 输入 token 的初始嵌入矩阵\nprint(token_embeddings.shape) # inputs是一个8x4矩阵（8批次4长度），则输出维度为[8, 4, 16]\n```\n输出\n\n```python\ntorch.Size([8, 4, 16])\n```\n\n\n- 位置嵌入（`Positional embedding layer`）可以理解为输入序列所有token之间的依赖关系在所有特征矩阵的初始值，代码如下：\n\n```python\ncontext_length = context_size # 输入上下文长度\npos_embedding_layer = torch.nn.Embedding(context_length, output_dim) # 初始化一个嵌入层\n\npos_embeddings = pos_embedding_layer(torch.arange(context_size)) # 输入序列位置嵌入的初始矩阵\nprint(pos_embeddings.shape) # 4x16\n```\n\n输出\n\n```python\ntorch.Size([4, 16])\n```\n\n\n然后将这两个相加可得嵌入层\n\n```python\ninput_embeddings = token_embeddings + pos_embeddings\nprint(input_embeddings.shape)\n```\n\n输出\n\n```python\ntorch.Size([8, 4, 16])\n```\n\n\n### 2.2 实现层归一化\n\n- 层归一化是将神经网络层的激活集中在平均值 0 附近，并将其方差归一化为 1，这可以稳定训练并更快地收敛到有效权重，\n- 应用于 Transformer 块中的多头注意模块之前和之后，也应用于最终输出层之前;\n- 一般通过减去平均值，并除以方差来执行标准化\n\n```python\nclass LayerNorm(nn.Module):\n    def __init__(self, emb_dim):\n        super().__init__()\n        self.eps = 1e-5\n        self.scale = nn.Parameter(torch.ones(emb_dim))\n        self.shift = nn.Parameter(torch.zeros(emb_dim))\n\n    def forward(self, x):\n        mean = x.mean(dim=-1, keepdim=True) # dim=-1意味着最后一个维度上计算平均值；keepdim=True，输出广播回原来的形状\n        var = x.var(dim=-1, keepdim=True, unbiased=False) # unbiased=False表示方差计算时除以样本大小（n），不包含矫正（n-1)，样本较大，忽略n和n-1的差异\n        norm_x = (x - mean) / torch.sqrt(var + self.eps)\n        return self.scale * norm_x + self.shift\n```\n\n - 实际实现时增加了一个较小的数 `eps` 避免方差为0时除以0的错误\n - 标准化时增加了两个参数 `scale` 和 `shift`，初始值分别为1和0，它们是可训练参数，在训练期间自动调整\n\n\n### 2.3 前馈网络\n\n前馈神经网络通过一系列线性变换和非线性激活函数对输入数据进行处理，以捕捉输入数据中的特征，并可能转换其表示形式。在此我们使用GELU非线性激活函数。通常实现一种计算上更便宜的近似值：$\\text{GELU}(x) \\approx 0.5 \\cdot x \\cdot \\left(1 + \\tanh\\left[\\sqrt{\\frac{2}{\\pi}} \\cdot \\left(x + 0.044715 \\cdot x^3\\right)\\right]\\right)\n$\n\n```python\nclass GELU(nn.Module):\n    def __init__(self):\n        super().__init__()\n\n    def forward(self, x):\n        return 0.5 * x * (1 + torch.tanh(torch.sqrt(torch.tensor(2.0 / torch.pi)) *(x + 0.044715 * torch.pow(x, 3))))\n```\n\n接下来，实现小型神经网络模块 `FeedForward`，使用线性层-激活函数-线性层的机构，代码如下：\n\n```python\nclass FeedForward(nn.Module):\n    def __init__(self, emb_dim):\n        super().__init__()\n        self.layers = nn.Sequential(\n            nn.Linear(emb_dim, 4 * emb_dim),\n            GELU(),\n            nn.Linear(4 * emb_dim, emb_dim)\n        )\n\n    def forward(self, x):\n        return self.layers(x)\n```\n\n### 2.4 跳跃连接\n跳跃连接也称为跳过或残差连接，用以缓解梯度消失问题，跳跃连接为梯度在网络中流动创建了一条替代的较短路径,这是通过将一层的输出添加到后一层的输出来实现的，通常会跳过中间的一个或多个层，可以看如下图：\n\n\u003cimg src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch04_compressed/12.webp?123\" width=\"400px\"\u003e\n\n### 2.5 多头注意力机制\n这种自注意力机制也称为“缩放点积注意力”，总体思路是将上下文向量计算为特定于某个输入元素的输入向量的加权和，初始有三个权重矩阵，模型训练期间更新权重矩阵\n\n#### 单头因果自注意力机制\n因果自注意力确保模型对序列中某个位置的预测仅依赖于先前位置的已知输出，而不依赖于未来位置，确保每个下一个单词的预测仅依赖于前面的单词\n\n- 首先引入三个训练权重矩阵 $W_q$、$W_k$ 和 $W_v$\n- **步骤 1** 这三个矩阵用于通过矩阵乘法将嵌入的输入标记 $x^{(i)}$ 投影到查询、键和值向量中：\n\n  - 查询向量：$q^{(i)} = W_q \\,x^{(i)}$\n  - 键向量：$k^{(i)} = W_k \\,x^{(i)}$\n  - 值向量：$v^{(i)} = W_v \\,x^{(i)}$\n\n- **步骤 2**，计算查询和每个关键向量之间的点积来计算非标准化注意力分数：\n  - 权重分数：$scores^{(i)} = q^{(i)} @ \\,k$\n  - 因果机制：对于每个给定的标记，屏蔽掉未来的标记，即输入文本中当前标记之后的标记\n\n- **步骤 3**，使用 softmax 函数计算注意力权重（总和为 1 的标准化注意力分数）,通过将注意力分数除以嵌入维度的平方根 $\\sqrt{d_k}$（即 `d_k**0.5`）来缩放注意力分数：\n  - 权重矩阵：$weights^{(i)} = softmax(scores^{(i)} / \\sqrt{d_k})$\n  - 掩盖权重：使用 dropout 掩盖额外的注意力权重\n\n- **步骤 4**，计算输入查询向量的上下文向量：\n  - 上下文向量：$context^{(i)} = weights^{(i)} @ \\,v$\n\n```python\nclass CausalSelfAttention(nn.Module):\n\n    def __init__(self, d_in, d_out, context_length,\n                 dropout, qkv_bias=False):\n        super().__init__()\n        self.d_out = d_out\n        self.W_query = nn.Linear(d_in, d_out, bias=qkv_bias)\n        self.W_key   = nn.Linear(d_in, d_out, bias=qkv_bias)\n        self.W_value = nn.Linear(d_in, d_out, bias=qkv_bias)\n        self.dropout = nn.Dropout(dropout) # New\n        self.register_buffer('mask', torch.triu(torch.ones(context_length, context_length), diagonal=1)) # New\n\n    def forward(self, x):\n        b, num_tokens, d_in = x.shape # New batch dimension b\n        keys = self.W_key(x)\n        queries = self.W_query(x)\n        values = self.W_value(x)\n\n        attn_scores = queries @ keys.transpose(1, 2) # Changed transpose\n        attn_scores.masked_fill_(  # New, _ ops are in-place\n            self.mask.bool()[:num_tokens, :num_tokens], -torch.inf)  # `:num_tokens` to account for cases where the number of tokens in the batch is smaller than the supported context_size\n        attn_weights = torch.softmax(\n            attn_scores / keys.shape[-1]**0.5, dim=-1\n        )\n        attn_weights = self.dropout(attn_weights) # New\n\n        context_vec = attn_weights @ values\n        return context_vec\n```\n\n#### 扩展到多头注意力\n多头注意力背后的主要思想是使用不同的、学习到的线性投影多次（并行）运行注意力机制。这允许模型共同关注来自不同位置的不同表示子空间的信息。只需堆叠多个单头注意力模块即可获得多头注意力模块：\n\n```python\nclass MultiHeadAttention(nn.Module):\n    def __init__(self, d_in, d_out, context_length, dropout, num_heads, qkv_bias=False):\n        super().__init__()\n        self.heads = nn.ModuleList(\n            [CausalSelfAttention(d_in, d_out, context_length, dropout, qkv_bias) \n             for _ in range(num_heads)]\n        )\n        self.out_proj = nn.Linear(d_out*num_heads, d_out*num_heads)\n\n    def forward(self, x):\n        context_vec = torch.cat([head(x) for head in self.heads], dim=-1)\n        return self.out_proj(context_vec)\n```\n\n虽然以上是多头注意力的直观且功能齐全的实现（包装了之前的单头注意力`CausalSelfAttention`实现），但我们可以编写一个名为`MultiHeadAttention`的独立类来实现相同的功能，创建单个W_query、W_key和W_value权重矩阵，然后将它们拆分为每个注意力头的单独矩阵：\n\n```python\n\nclass MultiHeadAttention(nn.Module):\n    def __init__(self, d_in, d_out, context_length, dropout, num_heads, qkv_bias=False):\n        super().__init__()\n        assert d_out % num_heads == 0, \"d_out must be divisible by num_heads\"\n\n        self.d_out = d_out\n        self.num_heads = num_heads\n        self.head_dim = d_out // num_heads  # Reduce the projection dim to match desired output dim\n\n        self.W_query = nn.Linear(d_in, d_out, bias=qkv_bias)\n        self.W_key = nn.Linear(d_in, d_out, bias=qkv_bias)\n        self.W_value = nn.Linear(d_in, d_out, bias=qkv_bias)\n        self.out_proj = nn.Linear(d_out, d_out)  # Linear layer to combine head outputs\n        self.dropout = nn.Dropout(dropout)\n        self.register_buffer('mask', torch.triu(torch.ones(context_length, context_length), diagonal=1))\n\n    def forward(self, x):\n        b, num_tokens, d_in = x.shape\n\n        keys = self.W_key(x)  # Shape: (b, num_tokens, d_out)\n        queries = self.W_query(x)\n        values = self.W_value(x)\n\n        # We implicitly split the matrix by adding a `num_heads` dimension\n        # Unroll last dim: (b, num_tokens, d_out) -\u003e (b, num_tokens, num_heads, head_dim)\n        keys = keys.view(b, num_tokens, self.num_heads, self.head_dim)\n        values = values.view(b, num_tokens, self.num_heads, self.head_dim)\n        queries = queries.view(b, num_tokens, self.num_heads, self.head_dim)\n\n        # Transpose: (b, num_tokens, num_heads, head_dim) -\u003e (b, num_heads, num_tokens, head_dim)\n        keys = keys.transpose(1, 2)\n        queries = queries.transpose(1, 2)\n        values = values.transpose(1, 2)\n\n        # Compute scaled dot-product attention (aka self-attention) with a causal mask\n        attn_scores = queries @ keys.transpose(2, 3)  # Dot product for each head\n\n        # Original mask truncated to the number of tokens and converted to boolean\n        mask_bool = self.mask.bool()[:num_tokens, :num_tokens]\n\n        # Use the mask to fill attention scores\n        attn_scores.masked_fill_(mask_bool, -torch.inf)\n\n        attn_weights = torch.softmax(attn_scores / keys.shape[-1]**0.5, dim=-1)\n        attn_weights = self.dropout(attn_weights)\n\n        # Shape: (b, num_tokens, num_heads, head_dim)\n        context_vec = (attn_weights @ values).transpose(1, 2)\n\n        # Combine heads, where self.d_out = self.num_heads * self.head_dim\n        context_vec = context_vec.contiguous().view(b, num_tokens, self.d_out)\n        context_vec = self.out_proj(context_vec)  # optional projection\n\n        return context_vec\n```\n\n### 2.6 实现 `Transformer`\n现在将前面的概念组合成所谓的 `transformer` 块，transformer 块还使用了 dropout 和快捷连接\n\n```python\nclass TransformerBlock(nn.Module):\n    def __init__(self, cfg):\n        super().__init__()\n        self.norm1 = LayerNorm(cfg[\"emb_dim\"])\n        self.att = MultiHeadAttention(\n            d_in=cfg[\"emb_dim\"],\n            # d_out=cfg[\"emb_dim\"]//cfg[\"n_heads\"], #如果使用堆叠多个头的方式，请将d_out赋值\n            d_out=cfg[\"emb_dim\"],\n            context_length=cfg[\"context_length\"],\n            num_heads=cfg[\"n_heads\"], \n            dropout=cfg[\"drop_rate\"],\n            qkv_bias=cfg[\"qkv_bias\"])\n        self.drop_shortcut = nn.Dropout(cfg[\"drop_rate\"])\n        self.norm2 = LayerNorm(cfg[\"emb_dim\"])\n        self.ff = FeedForward(cfg[\"emb_dim\"])\n\n    def forward(self, x):\n        # 注意力模块的残差连接\n        shortcut = x\n        x = self.norm1(x)\n        x = self.att(x)  # [batch_size, num_tokens, emb_size]\n        x = self.drop_shortcut(x)\n        x = x + shortcut  # 将原始输入添加回去\n\n        # 前馈模块的残差连接\n        shortcut = x\n        x = self.norm2(x)\n        x = self.ff(x)\n        x = self.drop_shortcut(x)\n        x = x + shortcut  # 将原始输入添加回去\n\n        return x\n```\n\n### 2.7 实现GPT模型\n最后将嵌入层、`transformer` 块、输出层组合以实现 `GPTModel`, 其中 `transformer` 块重复12次：\n\n```python\nclass GPTModel(nn.Module):\n    def __init__(self, cfg):\n        super().__init__()\n        self.tok_emb = nn.Embedding(cfg[\"vocab_size\"], cfg[\"emb_dim\"])\n        self.pos_emb = nn.Embedding(cfg[\"context_length\"], cfg[\"emb_dim\"])\n        self.drop_emb = nn.Dropout(cfg[\"drop_rate\"])\n        \n        self.trf_blocks = nn.Sequential(\n            *[TransformerBlock(cfg) for _ in range(cfg[\"n_layers\"])])\n        \n        self.final_norm = LayerNorm(cfg[\"emb_dim\"])\n        self.out_head = nn.Linear(\n            cfg[\"emb_dim\"], cfg[\"vocab_size\"], bias=False\n        )\n\n    def forward(self, in_idx):\n        batch_size, seq_len = in_idx.shape\n        tok_embeds = self.tok_emb(in_idx)\n        pos_embeds = self.pos_emb(torch.arange(seq_len, device=in_idx.device))\n        x = tok_embeds + pos_embeds  # Shape [batch_size, num_tokens, emb_size]\n        x = self.drop_emb(x)\n        x = self.trf_blocks(x)\n        x = self.final_norm(x)\n        logits = self.out_head(x)\n        return logits\n```\n\n- 使用模型的参数配置，可以随机初始权重实例化此 GPT 模型，如下所示：\n\n```python\ntorch.manual_seed(123)\nmodel = GPTModel(GPT_CONFIG_85M)\nout = model(inputs)\nprint(\"输入形状:\", inputs.shape)\nprint(\"输出形状:\", out.shape)\nprint(\"\\n\")\nprint(\"输入:\\n\", inputs)\nprint(\"输出:\\n\", out)\n```\n```python\n输入形状: torch.Size([8, 4])\n输出形状: torch.Size([8, 4, 323])\n\n\n输入:\n tensor([[199,   6, 210, 204],\n        [201, 298, 176, 184],\n        [188, 256, 185, 223],\n        [ 41,  45,   1,   1],\n        [ 87, 219,  80, 223],\n        [170, 293,   7, 319],\n        [145,  32, 199,  12],\n        [ 25, 298, 176, 255]])\n输出:\n tensor([[[ 0.3726,  0.0385,  0.0827,  ...,  0.4805,  0.9088, -0.6839],\n         [ 0.7950, -1.0513, -0.6098,  ..., -0.8869, -0.6006, -0.5492],\n         [ 0.9083, -0.9017, -0.3999,  ..., -0.3270, -0.2590,  0.2711],\n         [-1.0728, -0.3401,  0.8292,  ...,  0.4014, -1.2309,  0.2709]],\n\n        [[ 0.0749, -0.2326,  0.4346,  ...,  0.6214,  0.9596, -0.0181],\n         [-0.2092, -1.7445, -0.1894,  ..., -1.0994,  0.3208,  0.0739],\n         [ 0.0742, -1.2146, -0.7375,  ..., -0.7046, -0.2390,  0.9102],\n         [-0.4513, -0.9334,  0.5463,  ..., -0.3801, -1.3193,  0.1084]],\n\n        [[ 0.2166, -0.3299, -0.2181,  ...,  0.3662,  0.3434, -0.7779],\n...\n         [-0.3524, -1.3779, -0.1233,  ..., -1.4572, -0.0858, -0.5349],\n         [-0.2064, -1.5769, -0.7577,  ..., -0.7839, -0.3291,  0.6718],\n         [-0.3501, -0.6737,  0.1889,  ...,  0.0036, -0.7591, -0.4517]]],\n       grad_fn=\u003cUnsafeViewBackward0\u003e)\n```\n最后可以按如下方式计算模型的大小：\n\n```python\ntotal_params = sum(p.numel() for p in model.parameters())\nprint(f\"参数总数： {total_params:,}\")\n\ntotal_params_gpt2 =  total_params - sum(p.numel() for p in model.out_head.parameters())\nprint(f\"考虑到权重绑定的可训练参数数量： {total_params_gpt2:,}\")\n\n# Calculate the total size in bytes (assuming float32, 4 bytes per parameter)\ntotal_size_bytes = total_params * 4\n\n# Convert to megabytes\ntotal_size_mb = total_size_bytes / (1024 * 1024)\n\nprint(f\"所需内存大小: {total_size_mb:.2f} MB\")\n```\n```\n参数总数： 85,530,624\n考虑到权重绑定的可训练参数数量： 85,282,560\n所需内存大小: 326.27 MB\n```\n\n- 最后我们完善生成文本的方法 `generate_text_simple`，在每步预测结果中，获取最高的条目作为输出\n\n```python\ndef generate_text_simple(model, idx, max_new_tokens, context_size):\n    # idx 是当前上下文中的 (batch, n_tokens) 索引数组\n    for _ in range(max_new_tokens):\n\n        # 如果当前上下文超出支持的上下文大小，则裁剪当前上下文\n        # 例如，如果 LLM 仅支持 5 个 token，而上下文大小为 10\n        # 那么只有最后 5 个 token 用作上下文\n        idx_cond = idx[:, -context_size:] # 获取上下文token\n        \n        # 获取预测\n        with torch.no_grad():\n            logits = model(idx_cond) # 通过模型生成后续条目\n            \n        logits = logits[:, -1, :]   # 获取最后一个条目\n\n        probas = torch.softmax(logits, dim=-1) # 获取概率\n\n        idx_next = torch.argmax(probas, dim=-1, keepdim=True) # 选择概率最大的条目\n\n        idx = torch.cat((idx, idx_next), dim=1) #将下一个预测添加到序列中\n\n    return idx\n```\n\n让我们测试一下：\n\n```python\ntokenizer = SimpleTokenizer(vocab=create_vocab(raw_text=raw_text))\n\nstart_text = \"每一次努力都让你感动\"\nencoded = tokenizer.encode(start_text)\nprint(\"encoded:\", encoded)\n\nin_idx = torch.tensor(encoded).unsqueeze(0)\nprint(\"in_idx.shape:\", in_idx.shape)\n```\n\n```python\nencoded: [199, 6, 194, 55, 50, 298, 264, 38, 142, 53]\nin_idx.shape: torch.Size([1, 10])\n```\n\n```python\nmodel.eval() # disable dropout\n\nout = generate_text_simple(\n    model=model,\n    idx=in_idx, \n    max_new_tokens=15, \n    context_size=GPT_CONFIG_85M[\"context_length\"]\n)\n\nprint(\"输出:\", out)\nprint(\"输出长度:\", len(out[0]))\n```\n\n```python\n输出: tensor([[199,   6, 194,  55,  50, 298, 264,  38, 142,  53,  67,  11, 258,  54,\n         178, 159,  54, 178, 159,  54, 178, 159,  54, 178, 159]])\n输出长度: 25\n```\n\n删除批次维度并转换回文本：\n\n```python\ndecoded_text = tokenizer.decode(out.squeeze(0).tolist())\nprint(decoded_text)\n```\n```\n每一次努力都让你感动受且虹助普接助普接助普接助普接\n```\n\n请注意，该模型未经训练，因此上面的输出文本是随机的，下一节我们将训练这个模型。\n\n\n\n## 3. 模型训练与评估\n\n- 首先让我们把之前实现的代码归到一个模块中 `llm.py`，这样让下面的章节代码使用更简洁清晰一点\n- 然后将上一节最后测试时将文本转换为token ID 和 token ID转换为文本提炼为两个方法 `text_to_token_ids` 和 `token_ids_to_text`\n- 接着我们继续来看看前言中提出的问题，如何根据已有模型来预测下一个输出\n\n``` python\nfrom llm import SimpleTokenizer, create_vocab, GPTModel, generate_text_simple, text_to_token_ids, token_ids_to_text\n\nGPT_CONFIG_85M = {\n    \"vocab_size\": 323,      # 词汇表大小\n    \"context_length\": 8,    # 上下文长度\n    \"emb_dim\": 768,          # 嵌入维度\n    \"n_heads\": 12,          # 注意力头数量\n    \"n_layers\": 12,         # 层数\n    \"drop_rate\": 0.1,       # Dropout 率\n    \"qkv_bias\": False       # Query-Key-Value bias\n}\n\nwith open(\"the-road.txt\", \"r\", encoding=\"utf-8\") as f:\n    raw_text = f.read()\ntokenizer = SimpleTokenizer(vocab=create_vocab(raw_text=raw_text))\nmodel = GPTModel(cfg=GPT_CONFIG_85M)\n\nstart_text = \"每一次努力都让你感动\"\ntoken_ids = generate_text_simple(\n    model=model,\n    idx=text_to_token_ids(start_text, tokenizer),\n    max_new_tokens=10,\n    context_size=GPT_CONFIG_85M[\"context_length\"]\n)\n\nprint(\"输出文本:\\n\", token_ids_to_text(token_ids, tokenizer))\n```\n```python\n输出文本:\n 每一次努力都让你感动然然人耘定够场挥挥辉\n```\n\n### 3.1 训练集和验证集\n\n如上所示，由于模型尚未经过训练，因此无法生成优质文本。如何以数字形式测量或捕捉“优质文本”，以便在训练期间进行跟踪？\n- 其实，对于数值数据，我们可以测量两个数字之间的距离来标记它们之间的差异\n- 然后，假设我们有一组数据集，包含 `inputs` 张量，和其移动一个位置的 `targets`，这部分在数据预处理中已实现\n- 接着，我们可以将这个 `inputs` 输入到模型中，然后模型会输出对应的 `logits` 张量，这部分在 `generate_text_simple` 中已有相关实现\n- 最后，将 `targets` 和 `logits` 通过计算距离并对比来测量模型输出与正确的预测（目标）有多远\n- 在深度学习中，使用最大化概率分数的对数来表示距离目标有多远，这种测量方式也成为交叉熵损失\n- PyTorch 已经实现了一个 `cross_entropy` 函数来执行前面的步骤\n\n接下来，我们先将数据集分为训练和验证，然后使用交叉熵损失来计算损失\n\n```python\ntotal_characters = len(raw_text)\ntotal_tokens = len(tokenizer.encode(raw_text))\n\nprint(\"总字数:\", total_characters)\nprint(\"总标记数:\", total_tokens)\n```\n```python\n总字数: 778\n总标记数: 768\n```\n\n- 首先根据比例将数据集切割为训练集和验证集合：\n\n```python\nimport torch\nfrom llm import create_dataloader\n\n# 训练/验证比率\ntrain_ratio = 0.90\nsplit_idx = int(train_ratio * len(raw_text))\ntrain_data = raw_text[:split_idx]\nval_data = raw_text[split_idx:]\n\ntorch.manual_seed(123)\ntrain_loader = create_dataloader(\n    train_data,\n    vocab=create_vocab(raw_text=raw_text),\n    batch_size=2,\n    max_length=GPT_CONFIG_85M[\"context_length\"],\n    stride=GPT_CONFIG_85M[\"context_length\"],\n    drop_last=True,\n    shuffle=True,\n    num_workers=0\n)\n\nval_loader = create_dataloader(\n    val_data,\n    vocab=create_vocab(raw_text=raw_text),\n    batch_size=2,\n    max_length=GPT_CONFIG_85M[\"context_length\"],\n    stride=GPT_CONFIG_85M[\"context_length\"],\n    drop_last=False,\n    shuffle=False,\n    num_workers=0\n)\n```\n\n```python\n# 健全性检查\nif total_tokens * (train_ratio) \u003c GPT_CONFIG_85M[\"context_length\"]:\n    print(\"训练加载器没有足够的令牌。尝试降低 `GPT_CONFIG_85M['context_length']` 或 增加 `training_ratio`\")\n\nif total_tokens * (1-train_ratio) \u003c GPT_CONFIG_85M[\"context_length\"]:\n    print(\"验证加载器没有足够的令牌。尝试降低 `GPT_CONFIG_85M['context_length']` 或  降低 `training_ratio`\")\n```\n\n- 可选检查数据是否已正确加载：\n\n```python\nprint(\"训练集加载：\")\nfor x, y in train_loader:\n    print(x.shape, y.shape)\n\nprint(\"\\n验证集加载：\")\nfor x, y in val_loader:\n    print(x.shape, y.shape)\n```\n\n```python\n训练集加载：\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\n...\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([2, 8]) torch.Size([2, 8])\ntorch.Size([1, 8]) torch.Size([1, 8])\n```\n\n- 另一个可选检查是检查 `token` 数量是否在预期的范围内：\n\n```python\ntrain_tokens = 0\nfor input_batch, target_batch in train_loader:\n    train_tokens += input_batch.numel()\n\nval_tokens = 0\nfor input_batch, target_batch in val_loader:\n    val_tokens += input_batch.numel()\n\nprint(\"训练集 tokens:\", train_tokens)\nprint(\"验证集 tokens:\", val_tokens)\nprint(\"所有 tokens:\", train_tokens + val_tokens)\n```\n\n```python\n训练集 tokens: 688\n验证集 tokens: 72\n所有 tokens: 760\n```\n\n### 3.2 交叉熵损失\n\n- 我们实现一个方法来计算给定批次的交叉熵损失\n- 此外，我们实现第二个效用函数来计算数据加载器中用户指定批次数量的损失\n\n```python\ndef calc_loss_batch(input_batch, target_batch, model, device):\n    input_batch, target_batch = input_batch.to(device), target_batch.to(device)\n    logits = model(input_batch)\n    loss = torch.nn.functional.cross_entropy(logits.flatten(0, 1), target_batch.flatten())\n    return loss\n\n\ndef calc_loss_loader(data_loader, model, device, num_batches=None):\n    total_loss = 0.\n    if len(data_loader) == 0:\n        return float(\"nan\")\n    elif num_batches is None:\n        num_batches = len(data_loader)\n    else:\n        # 如果 num_batches 超过数据加载器中的批次数量，减少批次数量以匹配数据加载器中的批次总数\n        num_batches = min(num_batches, len(data_loader))\n    for i, (input_batch, target_batch) in enumerate(data_loader):\n        if i \u003c num_batches:\n            loss = calc_loss_batch(input_batch, target_batch, model, device)\n            total_loss += loss.item()\n        else:\n            break\n    return total_loss / num_batches\n```\n\n- 如果您的机器配有支持 CUDA 的 GPU，LLM 将在 GPU 上进行训练，而无需对代码进行任何更改\n- 通过 `device` 设置，我们确保数据加载到与 LLM 模型相同的设备上\n\n```python\nif torch.cuda.is_available():\n   device = torch.device(\"cuda\")\nelif torch.backends.mps.is_available():\n   device = torch.device(\"mps\")\nelse:\n   device = torch.device(\"cpu\")\n\nprint(f\"使用 {device} 设备.\")\n\nmodel.to(device) \n\ntorch.manual_seed(123) \n\nwith torch.no_grad():\n    train_loss = calc_loss_loader(train_loader, model, device)\n    val_loss = calc_loss_loader(val_loader, model, device)\n\nprint(\"训练集损失:\", train_loss)\nprint(\"验证集损失:\", val_loss)\n```\n\n```python\n使用 cpu 设备.\n训练集损失: 5.922613764918128\n验证集损失: 5.9078624725341795\n```\n\n### 3.3 训练大模型\n\n在本节我们实现一个简单的训练方法进行模型训练\n\n```python\ndef train_model_simple(model, train_loader, val_loader, optimizer, device, num_epochs,\n                       eval_freq, eval_iter, start_context, tokenizer):\n    # 初始化列表以追踪损失和看到的输出\n    train_losses, val_losses, track_tokens_seen = [], [], []\n    tokens_seen, global_step = 0, -1\n\n    # 主训练循环\n    for epoch in range(num_epochs):\n        model.train()  # # 将模型设置为训练模式\n        \n        for input_batch, target_batch in train_loader:\n            optimizer.zero_grad() # 重置上一次批次迭代的损失梯度\n            loss = calc_loss_batch(input_batch, target_batch, model, device)\n            loss.backward() # 计算损失梯度\n            optimizer.step() # 使用损失梯度更新模型权重\n            tokens_seen += input_batch.numel()\n            global_step += 1\n\n            # 可选评估步骤\n            if global_step % eval_freq == 0:\n                train_loss, val_loss = evaluate_model(\n                    model, train_loader, val_loader, device, eval_iter)\n                train_losses.append(train_loss)\n                val_losses.append(val_loss)\n                track_tokens_seen.append(tokens_seen)\n                print(f\"Ep {epoch+1} (Step {global_step:06d}): \"\n                      f\"训练损失 {train_loss:.3f}, 验证损失 {val_loss:.3f}\")\n\n        # 每个时期后打印一个示例文本\n        generate_and_print_sample(\n            model, tokenizer, device, start_context\n        )\n\n    return train_losses, val_losses, track_tokens_seen\n\n\ndef evaluate_model(model, train_loader, val_loader, device, eval_iter):\n    model.eval()\n    with torch.no_grad():\n        train_loss = calc_loss_loader(train_loader, model, device, num_batches=eval_iter)\n        val_loss = calc_loss_loader(val_loader, model, device, num_batches=eval_iter)\n    model.train()\n    return train_loss, val_loss\n\n\ndef generate_and_print_sample(model, tokenizer, device, start_context):\n    model.eval()\n    context_size = model.pos_emb.weight.shape[0]\n    encoded = text_to_token_ids(start_context, tokenizer).to(device)\n    with torch.no_grad():\n        token_ids = generate_text_simple(\n            model=model, idx=encoded,\n            max_new_tokens=50, context_size=context_size\n        )\n    decoded_text = token_ids_to_text(token_ids, tokenizer)\n    print(decoded_text.replace(\"\\n\", \" \"))  # Compact print format\n    model.train()\n```\n\n- 现在，让我们使用上面定义的训练函数来训练 LLM：\n\n```python\nimport time\nstart_time = time.time()\n\ntorch.manual_seed(123)\nmodel = GPTModel(GPT_CONFIG_85M)\nmodel.to(device)\noptimizer = torch.optim.AdamW(model.parameters(), lr=0.0004, weight_decay=0.1)\n\nnum_epochs = 10\ntrain_losses, val_losses, tokens_seen = train_model_simple(\n    model, train_loader, val_loader, optimizer, device,\n    num_epochs=num_epochs, eval_freq=5, eval_iter=5,\n    start_context=\"每一次努力都让你感动\", tokenizer=tokenizer\n)\n\nend_time = time.time()\nexecution_time_minutes = (end_time - start_time) / 60\nprint(f\"训练在 {execution_time_minutes:.2f} 分钟内完成。\")\n```\n\n```python\nEp 1 (Step 000000): 训练损失 6.495, 验证损失 6.622\nEp 1 (Step 000005): 训练损失 5.938, 验证损失 6.079\nEp 1 (Step 000010): 训练损失 5.724, 验证损失 6.265\nEp 1 (Step 000015): 训练损失 5.127, 验证损失 6.435\nEp 1 (Step 000020): 训练损失 5.420, 验证损失 6.556\nEp 1 (Step 000025): 训练损失 5.566, 验证损失 6.544\nEp 1 (Step 000030): 训练损失 5.482, 验证损失 6.460\nEp 1 (Step 000035): 训练损失 5.352, 验证损失 6.386\nEp 1 (Step 000040): 训练损失 4.971, 验证损失 6.757\n每一次努力都让你感动的。我们的。我们的。。我们的。我们的。。我们的。我们的。。我们的。我们的。。我们的。我们的。。我们的\nEp 2 (Step 000045): 训练损失 5.745, 验证损失 6.569\nEp 2 (Step 000050): 训练损失 5.352, 验证损失 6.515\nEp 2 (Step 000055): 训练损失 4.909, 验证损失 6.761\nEp 2 (Step 000060): 训练损失 5.338, 验证损失 6.810\nEp 2 (Step 000065): 训练损失 4.773, 验证损失 6.682\nEp 2 (Step 000070): 训练损失 4.298, 验证损失 6.594\nEp 2 (Step 000075): 训练损失 4.797, 验证损失 6.583\nEp 2 (Step 000080): 训练损失 4.564, 验证损失 6.882\nEp 2 (Step 000085): 训练损失 4.333, 验证损失 6.319\n每一次努力都让你感动中来的中中来的中来的中中来的中来的中中来的中来的中中来的中来的中中来的中来的中中来的中来的中中来的中\nEp 3 (Step 000090): 训练损失 4.717, 验证损失 5.658\nEp 3 (Step 000095): 训练损失 4.326, 验证损失 6.135\nEp 3 (Step 000100): 训练损失 4.390, 验证损失 6.231\nEp 3 (Step 000105): 训练损失 4.148, 验证损失 6.353\nEp 3 (Step 000110): 训练损失 3.734, 验证损失 6.314\n...\nEp 10 (Step 000420): 训练损失 0.163, 验证损失 6.890\nEp 10 (Step 000425): 训练损失 0.149, 验证损失 6.898\n每一次努力都让你感动。当我们的努力。每一点小，我们也是学习毅加实验有了梦想的努力。每一点小，我们也是学习毅加实验有了梦想\n训练在 2.87 分钟内完成。\n```\n将上述的训练和验证损失绘制到图中查看结果：\n\n```python\nimport matplotlib.pyplot as plt\nfrom matplotlib.ticker import MaxNLocator\n\n\ndef plot_losses(epochs_seen, tokens_seen, train_losses, val_losses):\n    fig, ax1 = plt.subplots(figsize=(5, 3))\n\n    # Plot training and validation loss against epochs\n    ax1.plot(epochs_seen, train_losses, label=\"Training loss\")\n    ax1.plot(epochs_seen, val_losses, linestyle=\"-.\", label=\"Validation loss\")\n    ax1.set_xlabel(\"Epochs\")\n    ax1.set_ylabel(\"Loss\")\n    ax1.legend(loc=\"upper right\")\n    ax1.xaxis.set_major_locator(MaxNLocator(integer=True))  # only show integer labels on x-axis\n\n    # Create a second x-axis for tokens seen\n    ax2 = ax1.twiny()  # Create a second x-axis that shares the same y-axis\n    ax2.plot(tokens_seen, train_losses, alpha=0)  # Invisible plot for aligning ticks\n    ax2.set_xlabel(\"Tokens seen\")\n\n    fig.tight_layout()  # Adjust layout to make room\n    plt.savefig(\"loss-plot.pdf\")\n    plt.show()\n\nepochs_tensor = torch.linspace(0, num_epochs, len(train_losses))\nplot_losses(epochs_tensor, tokens_seen, train_losses, val_losses)\n```\n\n![alt text](image.png)\n\n- 查看上面的结果，我们可以看到模型一开始会生成难以理解的字符串，而到最后，它能够生成语法或多或少正确的句子\n- 但是，根据训练和验证集损失，我们可以看到模型开始过度拟合\n- 如果我们检查它在最后写的几段话，我们会发现它们逐字逐句地包含在训练集中——它只是记住了训练数据\n- 稍后，我们将介绍可以在一定程度上减轻这种记忆的解码策略\n- 请注意，这里发生过度拟合是因为我们有一个非常非常小的训练集，并且我们对其进行了多次迭代\n- 这里的 LLM 培训主要用于学习目的；我们主要想看看模型可以学会生成连贯的文本\n\n### 3.4 保存模型权重\n\n- 训练 LLM 的计算成本很高，因此能够保存和加载 LLM 权重至关重要\n- PyTorch 中推荐的方式是通过将 `torch.save` 函数应用于 `.state_dict()` 方法来保存模型权重，即所谓的 `state_dict`\n- 通常使用 Adam 或 AdamW 等自适应优化器来训练，这些自适应优化器会为每个模型权重存储额外的参数，因此如果我们计划稍后继续进行预训练，那么保存这些参数也是有意义的：\n\n```python\ntorch.save({\n    \"model_state_dict\": model.state_dict(),\n    \"optimizer_state_dict\": optimizer.state_dict(),\n    }, \n    \"model_and_optimizer.pth\"\n)\n```\n\n加载的代码如下：\n\n```python\ncheckpoint = torch.load(\"model_and_optimizer.pth\", weights_only=True)\n\nmodel = GPTModel(GPT_CONFIG_85M)\nmodel.load_state_dict(checkpoint[\"model_state_dict\"])\n\noptimizer = torch.optim.AdamW(model.parameters(), lr=0.0005, weight_decay=0.1)\noptimizer.load_state_dict(checkpoint[\"optimizer_state_dict\"])\nmodel.train();\n```\n\n## 4. 文本生成\n\n- 首先将上一节关于模型训练的代码移动到 `llm.py`\n- 使用我们之前在简单训练函数中使用的 `generate_text_simple` 函数，可以一次生成一个标记的新文本\n- 下一个生成的标记是词汇表中所有标记中概率得分最大的标记\n\n```python\nfrom llm import SimpleTokenizer, create_vocab, GPTModel, generate_text_simple, text_to_token_ids, token_ids_to_text\n\nGPT_CONFIG_85M = {\n    \"vocab_size\": 323,      # 词汇表大小\n    \"context_length\": 8,    # 上下文长度\n    \"emb_dim\": 768,          # 嵌入维度\n    \"n_heads\": 12,          # 注意力头数量\n    \"n_layers\": 12,         # 层数\n    \"drop_rate\": 0.1,       # Dropout 率\n    \"qkv_bias\": False       # Query-Key-Value bias\n}\n\n\nwith open(\"the-road.txt\", \"r\", encoding=\"utf-8\") as f:\n    raw_text = f.read()\n\ntokenizer = SimpleTokenizer(vocab=create_vocab(raw_text=raw_text))\n\n# 加载模型\ncheckpoint = torch.load(\"model_and_optimizer.pth\", weights_only=True)\nmodel = GPTModel(GPT_CONFIG_85M)\nmodel.load_state_dict(checkpoint[\"model_state_dict\"])\noptimizer = torch.optim.AdamW(model.parameters(), lr=0.0005, weight_decay=0.1)\noptimizer.load_state_dict(checkpoint[\"optimizer_state_dict\"])\nmodel.to(\"cpu\")\nmodel.eval()\n\nstart_text = \"每一次努力都让你感动\"\ntoken_ids = generate_text_simple(\n    model=model,\n    idx=text_to_token_ids(start_text, tokenizer),\n    max_new_tokens=25,\n    context_size=GPT_CONFIG_85M[\"context_length\"]\n)\n\nprint(\"输出文本:\\n\", token_ids_to_text(token_ids, tokenizer))\n```\n\n```python\n输出文本:\n 每一次努力都让你感动。当我们的努力。每一点小，我们也是学习毅加实验有了\n```\n\n即使我们多次执行上面的生成文本函数，LLM 也始终会生成相同的输出。我们现在引入两个概念，即所谓的解码策略，来修改 `generate_text_simple`：*温度缩放* 和 *top-k* 采样，这些将允许模型控制生成文本的随机性和多样性。\n- 温度缩放：使用 `torch.multinomial(probs, num_samples=1)` 从概率分布中采样下一个标记\n- `Top-K` 采样：为了能够使用更高的温度来增加输出多样性并降低无意义句子的概率，我们可以将采样的标记限制为前 k 个最可能的标记：\n\n\n```python\ndef generate(model, idx, max_new_tokens, context_size, temperature=0.0, top_k=None, eos_id=None):\n\n    # For 循环与之前相同：获取 logits，并且仅关注最后一个时间步\n    for _ in range(max_new_tokens):\n        idx_cond = idx[:, -context_size:]\n        with torch.no_grad():\n            logits = model(idx_cond)\n        logits = logits[:, -1, :]\n\n        # 新功能：使用 top_k 采样过滤 logits\n        if top_k is not None:\n            # 仅保留top_k个值\n            top_logits, _ = torch.topk(logits, top_k)\n            min_val = top_logits[:, -1]\n            logits = torch.where(logits \u003c min_val, torch.tensor(float(\"-inf\")).to(logits.device), logits)\n\n        # 新功能：应用温度缩放\n        if temperature \u003e 0.0:\n            logits = logits / temperature\n\n            # 应用 softmax 获取概率\n            probs = torch.softmax(logits, dim=-1)  # (batch_size, context_len)\n\n            # 从分布中抽样\n            idx_next = torch.multinomial(probs, num_samples=1)  # (batch_size, 1)\n\n        # 否则与之前相同：获取具有最高 logits 值的词汇条目的 idx\n        else:\n            idx_next = torch.argmax(logits, dim=-1, keepdim=True)  # (batch_size, 1)\n\n        if idx_next == eos_id:  # 如果遇到序列结束标记并且指定了 eos_id，则提前停止生成\n            break\n\n        # 与之前相同：将采样索引附加到运行序列\n        idx = torch.cat((idx, idx_next), dim=1)  # (batch_size, num_tokens+1)\n\n    return idx\n```\n\n```python\ntorch.manual_seed(123)\n\ntoken_ids = generate(\n    model=model,\n    idx=text_to_token_ids(\"每一次努力都让你感动\", tokenizer),\n    max_new_tokens=50,\n    context_size=GPT_CONFIG_85M[\"context_length\"],\n    top_k=25,\n    temperature=1.4\n)\n\nprint(\"Output text:\\n\", token_ids_to_text(token_ids, tokenizer))\n```\n\n```python\nOutput text:\n 每一次努力都让你感动。每一这个过我们也是学习间最家从中吸战付出努力。相此，以，以，用为边心怀感恩之最并学学习的财富终习。\n```\n\n## 结语\n\n最后，我们完成了一个简单版的大模型，包括文本数据预处理、模型架构实现、模型训练和文本生成等，最后我们使用完整的代码来演示一下文本生成：\n\n```python\nfrom llm import SimpleTokenizer, create_vocab, GPTModel, generate, text_to_token_ids, token_ids_to_text\n\nwith open(\"the-road.txt\", \"r\", encoding=\"utf-8\") as f:\n    raw_text = f.read()\n\nGPT_CONFIG_85M = {\n    \"vocab_size\": 323,      # 词汇表大小\n    \"context_length\": 8,    # 上下文长度\n    \"emb_dim\": 768,          # 嵌入维度\n    \"n_heads\": 12,          # 注意力头数量\n    \"n_layers\": 12,         # 层数\n    \"drop_rate\": 0.1,       # Dropout 率\n    \"qkv_bias\": False       # Query-Key-Value bias\n}\n\ntokenizer = SimpleTokenizer(vocab=create_vocab(raw_text=raw_text))\n\n# 加载模型\ncheckpoint = torch.load(\"model_and_optimizer.pth\", weights_only=True)\nmodel = GPTModel(GPT_CONFIG_85M)\nmodel.load_state_dict(checkpoint[\"model_state_dict\"])\noptimizer = torch.optim.AdamW(model.parameters(), lr=0.0005, weight_decay=0.1)\noptimizer.load_state_dict(checkpoint[\"optimizer_state_dict\"])\nmodel.to(\"cpu\")\nmodel.eval()\n\nstart_text = \"每一次努力都让你感动\"\ntoken_ids = generate(\n    model=model,\n    idx=text_to_token_ids(start_text, tokenizer),\n    max_new_tokens=25,\n    context_size=GPT_CONFIG_85M[\"context_length\"]\n)\n\nprint(\"输出文本:\\n\", token_ids_to_text(token_ids, tokenizer))\n```\n```python\n输出文本:\n 每一次努力都让你感动。当我们的努力。每一点小，我们也是学习毅加实验有了\n ```\n\n从输出的结果来看，内容也不尽人意，当然这个是因为我们的训练集太小了，我们可以选择一些公共数据集进行训练，以取得更好的结果，生成优质的文本，但这也需要更多的资源和时间。\n\n另外，在最后我附上本文参考的内容：[Build a Large Language Model (From Scratch)](https://github.com/rasbt/LLMs-from-scratch)\n\n\n## License\n\n本项目采用 MIT 许可证。根据此许可证，您可以在遵循 MIT 许可证条款的前提下自由使用、修改和分发本仓库中的内容。然而，如果您打算将本项目的任何部分用于商业用途，请务必遵守以下额外条件：\n\n- **标明原作者**：请在您的作品或产品中明确标注原作者，并附上原作者的名字。\n- **提供链接**：请提供一个指向本 GitHub 仓库的链接，以便用户能够找到原始项目。\n\n有关 MIT 许可证的更多信息，请参阅 [MIT 许可证](LICENSE) 文件。","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fgavinhome%2Fllm-from-zero","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fgavinhome%2Fllm-from-zero","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fgavinhome%2Fllm-from-zero/lists"}