diff --git a/ch02/02_bonus_bytepair-encoder/.ipynb_checkpoints/bpe_openai_gpt2-checkpoint.py b/ch02/02_bonus_bytepair-encoder/.ipynb_checkpoints/bpe_openai_gpt2-checkpoint.py deleted file mode 100644 index f3d9575..0000000 --- a/ch02/02_bonus_bytepair-encoder/.ipynb_checkpoints/bpe_openai_gpt2-checkpoint.py +++ /dev/null @@ -1,174 +0,0 @@ -""" -Byte pair encoding utilities - -Code from https://github.com/openai/gpt-2/blob/master/src/encoder.py - -And modified code (download_vocab) from -https://github.com/openai/gpt-2/blob/master/download_model.py - -Modified MIT License - -Software Copyright (c) 2019 OpenAI - -We don’t claim ownership of the content you create with GPT-2, so it is yours to do with as you please. -We only ask that you use GPT-2 responsibly and clearly indicate your content was created using GPT-2. - -Permission is hereby granted, free of charge, to any person obtaining a copy of this software and -associated documentation files (the "Software"), to deal in the Software without restriction, -including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, -and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, -subject to the following conditions: - -The above copyright notice and this permission notice shall be included -in all copies or substantial portions of the Software. -The above copyright notice and this permission notice need not be included -with content created by the Software. - -THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, -INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, -FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS -BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, -TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE -OR OTHER DEALINGS IN THE SOFTWARE. - - -""" - -import os -import json -import regex as re -import requests -from tqdm import tqdm -from functools import lru_cache - -@lru_cache() -def bytes_to_unicode(): - """ - Returns list of utf-8 byte and a corresponding list of unicode strings. - The reversible bpe codes work on unicode strings. - This means you need a large # of unicode characters in your vocab if you want to avoid UNKs. - When you're at something like a 10B token dataset you end up needing around 5K for decent coverage. - This is a significant percentage of your normal, say, 32K bpe vocab. - To avoid that, we want lookup tables between utf-8 bytes and unicode strings. - And avoids mapping to whitespace/control characters the bpe code barfs on. - """ - bs = list(range(ord("!"), ord("~")+1))+list(range(ord("¡"), ord("¬")+1))+list(range(ord("®"), ord("ÿ")+1)) - cs = bs[:] - n = 0 - for b in range(2**8): - if b not in bs: - bs.append(b) - cs.append(2**8+n) - n += 1 - cs = [chr(n) for n in cs] - return dict(zip(bs, cs)) - -def get_pairs(word): - """Return set of symbol pairs in a word. - - Word is represented as tuple of symbols (symbols being variable-length strings). - """ - pairs = set() - prev_char = word[0] - for char in word[1:]: - pairs.add((prev_char, char)) - prev_char = char - return pairs - -class Encoder: - def __init__(self, encoder, bpe_merges, errors='replace'): - self.encoder = encoder - self.decoder = {v:k for k,v in self.encoder.items()} - self.errors = errors # how to handle errors in decoding - self.byte_encoder = bytes_to_unicode() - self.byte_decoder = {v:k for k, v in self.byte_encoder.items()} - self.bpe_ranks = dict(zip(bpe_merges, range(len(bpe_merges)))) - self.cache = {} - - # Should haved added re.IGNORECASE so BPE merges can happen for capitalized versions of contractions - self.pat = re.compile(r"""'s|'t|'re|'ve|'m|'ll|'d| ?\p{L}+| ?\p{N}+| ?[^\s\p{L}\p{N}]+|\s+(?!\S)|\s+""") - - def bpe(self, token): - if token in self.cache: - return self.cache[token] - word = tuple(token) - pairs = get_pairs(word) - - if not pairs: - return token - - while True: - bigram = min(pairs, key = lambda pair: self.bpe_ranks.get(pair, float('inf'))) - if bigram not in self.bpe_ranks: - break - first, second = bigram - new_word = [] - i = 0 - while i < len(word): - try: - j = word.index(first, i) - new_word.extend(word[i:j]) - i = j - except: - new_word.extend(word[i:]) - break - - if word[i] == first and i < len(word)-1 and word[i+1] == second: - new_word.append(first+second) - i += 2 - else: - new_word.append(word[i]) - i += 1 - new_word = tuple(new_word) - word = new_word - if len(word) == 1: - break - else: - pairs = get_pairs(word) - word = ' '.join(word) - self.cache[token] = word - return word - - def encode(self, text): - bpe_tokens = [] - for token in re.findall(self.pat, text): - token = ''.join(self.byte_encoder[b] for b in token.encode('utf-8')) - bpe_tokens.extend(self.encoder[bpe_token] for bpe_token in self.bpe(token).split(' ')) - return bpe_tokens - - def decode(self, tokens): - text = ''.join([self.decoder[token] for token in tokens]) - text = bytearray([self.byte_decoder[c] for c in text]).decode('utf-8', errors=self.errors) - return text - -def get_encoder(model_name, models_dir): - with open(os.path.join(models_dir, model_name, 'encoder.json'), 'r') as f: - encoder = json.load(f) - with open(os.path.join(models_dir, model_name, 'vocab.bpe'), 'r', encoding="utf-8") as f: - bpe_data = f.read() - bpe_merges = [tuple(merge_str.split()) for merge_str in bpe_data.split('\n')[1:-1]] - return Encoder( - encoder=encoder, - bpe_merges=bpe_merges, - ) - - -def download_vocab(): - # Modified code from - subdir = 'gpt2_model' - if not os.path.exists(subdir): - os.makedirs(subdir) - subdir = subdir.replace('\\','/') # needed for Windows - - for filename in ['encoder.json', 'vocab.bpe']: - - r = requests.get("https://openaipublic.blob.core.windows.net/gpt-2/models/117M" + "/" + filename, stream=True) - - with open(os.path.join(subdir, filename), 'wb') as f: - file_size = int(r.headers["content-length"]) - chunk_size = 1000 - with tqdm(ncols=100, desc="Fetching " + filename, total=file_size, unit_scale=True) as pbar: - # 1k for chunk_size, since Ethernet packet size is around 1500 bytes - for chunk in r.iter_content(chunk_size=chunk_size): - f.write(chunk) - pbar.update(chunk_size) diff --git a/ch02/02_bonus_bytepair-encoder/.ipynb_checkpoints/compare-bpe-tiktoken-checkpoint.ipynb b/ch02/02_bonus_bytepair-encoder/.ipynb_checkpoints/compare-bpe-tiktoken-checkpoint.ipynb deleted file mode 100644 index 7448afb..0000000 --- a/ch02/02_bonus_bytepair-encoder/.ipynb_checkpoints/compare-bpe-tiktoken-checkpoint.ipynb +++ /dev/null @@ -1,442 +0,0 @@ -{ - "cells": [ - { - "cell_type": "markdown", - "id": "a9adc3bf-353c-411e-a471-0e92786e7103", - "metadata": {}, - "source": [ - "# Using BytePair encodding from `tiktoken`" - ] - }, - { - "cell_type": "code", - "execution_count": 1, - "id": "4036ffa3-0e5c-433a-a997-4ed7d33de0b2", - "metadata": {}, - "outputs": [], - "source": [ - "# !pip install tiktoken" - ] - }, - { - "cell_type": "code", - "execution_count": 4, - "id": "1c490fca-a48a-47fa-a299-322d1a08ad17", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "tiktoken version: 0.5.2\n" - ] - } - ], - "source": [ - "import importlib.metadata\n", - "\n", - "print(\"tiktoken version:\", importlib.metadata.version(\"tiktoken\"))" - ] - }, - { - "cell_type": "code", - "execution_count": 5, - "id": "0952667c-ce84-4f21-87db-59f52b44cec4", - "metadata": {}, - "outputs": [], - "source": [ - "import tiktoken\n", - "\n", - "tik_tokenizer = tiktoken.get_encoding(\"gpt2\")\n", - "\n", - "text = \"Hello, world. Is this-- a test?\"" - ] - }, - { - "cell_type": "code", - "execution_count": 6, - "id": "b039c350-18ad-48fb-8e6a-085702dfc330", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "[15496, 11, 995, 13, 1148, 428, 438, 257, 1332, 30]\n" - ] - } - ], - "source": [ - "integers = tik_tokenizer.encode(text, allowed_special={\"<|endoftext|>\"})\n", - "\n", - "print(integers)" - ] - }, - { - "cell_type": "code", - "execution_count": 7, - "id": "7b152ba4-04d3-41cc-849f-adedcfb8cabb", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Hello, world. Is this-- a test?\n" - ] - } - ], - "source": [ - "strings = tik_tokenizer.decode(integers)\n", - "\n", - "print(strings)" - ] - }, - { - "cell_type": "code", - "execution_count": 8, - "id": "cf148a1a-316b-43ec-b7ba-1b6d409ce837", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "50257\n" - ] - } - ], - "source": [ - "print(tik_tokenizer.n_vocab)" - ] - }, - { - "cell_type": "markdown", - "id": "6a0b5d4f-2af9-40de-828c-063c4243e771", - "metadata": {}, - "source": [ - "# Using the original Byte-pair encoding implementation used in GPT-2" - ] - }, - { - "cell_type": "code", - "execution_count": 9, - "id": "0903108c-65cb-4ae1-967a-2155e25349c2", - "metadata": {}, - "outputs": [], - "source": [ - "from bpe_openai_gpt2 import get_encoder, download_vocab" - ] - }, - { - "cell_type": "code", - "execution_count": 10, - "id": "35dd8d7c-8c12-4b68-941a-0fd05882dd45", - "metadata": {}, - "outputs": [ - { - "name": "stderr", - "output_type": "stream", - "text": [ - "Fetching encoder.json: 1.04Mit [00:28, 36.8kit/s] \n", - "Fetching vocab.bpe: 457kit [00:00, 458kit/s] \n" - ] - } - ], - "source": [ - "download_vocab()" - ] - }, - { - "cell_type": "code", - "execution_count": 14, - "id": "1888a7a9-9c40-4fe0-99b4-ebd20aa1ffd0", - "metadata": {}, - "outputs": [], - "source": [ - "orig_tokenizer = get_encoder(model_name=\"gpt2_model\", models_dir=\".\")" - ] - }, - { - "cell_type": "code", - "execution_count": 15, - "id": "2740510c-a78a-4fba-ae18-2b156ba2dfef", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "[15496, 11, 995, 13, 1148, 428, 438, 257, 1332, 30]\n" - ] - } - ], - "source": [ - "integers = orig_tokenizer.encode(text)\n", - "\n", - "print(integers)" - ] - }, - { - "cell_type": "code", - "execution_count": 16, - "id": "434d115e-990d-42ad-88dd-31323a96e10f", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Hello, world. Is this-- a test?\n" - ] - } - ], - "source": [ - "strings = orig_tokenizer.decode(integers)\n", - "\n", - "print(strings)" - ] - }, - { - "cell_type": "markdown", - "id": "4f63e8c6-707c-4d66-bcf8-dd790647cc86", - "metadata": {}, - "source": [ - "# Using the BytePair Tokenizer in HuggingFace transformers" - ] - }, - { - "cell_type": "code", - "execution_count": 12, - "id": "5bfff386-f725-4137-9c50-e5da0c38bea0", - "metadata": {}, - "outputs": [], - "source": [ - "# pip install transformers" - ] - }, - { - "cell_type": "code", - "execution_count": 13, - "id": "e9077bf4-f91f-42ad-ab76-f3d89128510e", - "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "'4.30.2'" - ] - }, - "execution_count": 13, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "import transformers\n", - "\n", - "transformers.__version__" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "16e06ee5-c4ca-4211-8e24-dbfd84b1d85b", - "metadata": {}, - "outputs": [], - "source": [ - "设置为国内可访问" - ] - }, - { - "cell_type": "code", - "execution_count": 1, - "id": "3e07ddc9-187e-4482-a7b5-7e4e9381d805", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "env: HF_ENDPOINT=https://hf-mirror.com\n" - ] - } - ], - "source": [ - "%env HF_ENDPOINT=https://hf-mirror.com" - ] - }, - { - "cell_type": "code", - "execution_count": 2, - "id": "a9839137-b8ea-4a2c-85fc-9a63064cf8c8", - "metadata": {}, - "outputs": [ - { - "data": { - "application/vnd.jupyter.widget-view+json": { - "model_id": "afc151b540664287aa60a4cbe90cdfeb", - "version_major": 2, - "version_minor": 0 - }, - "text/plain": [ - "vocab.json: 0.00B [00:00, ?B/s]" - ] - }, - "metadata": {}, - "output_type": "display_data" - }, - { - "data": { - "application/vnd.jupyter.widget-view+json": { - "model_id": "9a5d584e4adf40bca215b409b693dc02", - "version_major": 2, - "version_minor": 0 - }, - "text/plain": [ - "merges.txt: 0.00B [00:00, ?B/s]" - ] - }, - "metadata": {}, - "output_type": "display_data" - }, - { - "data": { - "application/vnd.jupyter.widget-view+json": { - "model_id": "a126ee77a9f94e58b1dcccd68e6d5bb1", - "version_major": 2, - "version_minor": 0 - }, - "text/plain": [ - "config.json: 0%| | 0.00/367 [00:00\"})\n", "\n", "print(integers)" @@ -74,7 +76,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 4, "id": "7b152ba4-04d3-41cc-849f-adedcfb8cabb", "metadata": {}, "outputs": [ @@ -87,6 +89,7 @@ } ], "source": [ + "# 进行解码\n", "strings = tik_tokenizer.decode(integers)\n", "\n", "print(strings)" @@ -94,7 +97,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": 5, "id": "cf148a1a-316b-43ec-b7ba-1b6d409ce837", "metadata": {}, "outputs": [ @@ -107,6 +110,7 @@ } ], "source": [ + "# 表示编码器的词汇表大小\n", "print(tik_tokenizer.n_vocab)" ] }, @@ -115,12 +119,12 @@ "id": "6a0b5d4f-2af9-40de-828c-063c4243e771", "metadata": {}, "source": [ - "# Using the original Byte-pair encoding implementation used in GPT-2" + "# 使用在GPT-2中使用的原始字节对编码实现" ] }, { "cell_type": "code", - "execution_count": 9, + "execution_count": 6, "id": "0903108c-65cb-4ae1-967a-2155e25349c2", "metadata": {}, "outputs": [], @@ -130,7 +134,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": 7, "id": "35dd8d7c-8c12-4b68-941a-0fd05882dd45", "metadata": {}, "outputs": [ @@ -138,8 +142,8 @@ "name": "stderr", "output_type": "stream", "text": [ - "Fetching encoder.json: 1.04Mit [00:28, 36.8kit/s] \n", - "Fetching vocab.bpe: 457kit [00:00, 458kit/s] \n" + "Fetching encoder.json: 1.04Mit [00:02, 502kit/s] \n", + "Fetching vocab.bpe: 457kit [00:02, 212kit/s] \n" ] } ], @@ -149,7 +153,7 @@ }, { "cell_type": "code", - "execution_count": 14, + "execution_count": 8, "id": "1888a7a9-9c40-4fe0-99b4-ebd20aa1ffd0", "metadata": {}, "outputs": [], @@ -159,7 +163,7 @@ }, { "cell_type": "code", - "execution_count": 15, + "execution_count": 9, "id": "2740510c-a78a-4fba-ae18-2b156ba2dfef", "metadata": {}, "outputs": [ @@ -179,7 +183,7 @@ }, { "cell_type": "code", - "execution_count": 16, + "execution_count": 10, "id": "434d115e-990d-42ad-88dd-31323a96e10f", "metadata": {}, "outputs": [ @@ -202,12 +206,14 @@ "id": "4f63e8c6-707c-4d66-bcf8-dd790647cc86", "metadata": {}, "source": [ - "# Using the BytePair Tokenizer in HuggingFace transformers" + "# 使用HuggingFace Transformers中的BytePair Tokenizer\n", + "\r\n", + "\r\n" ] }, { "cell_type": "code", - "execution_count": 12, + "execution_count": 11, "id": "5bfff386-f725-4137-9c50-e5da0c38bea0", "metadata": {}, "outputs": [], @@ -217,17 +223,17 @@ }, { "cell_type": "code", - "execution_count": 13, + "execution_count": 12, "id": "e9077bf4-f91f-42ad-ab76-f3d89128510e", "metadata": {}, "outputs": [ { "data": { "text/plain": [ - "'4.30.2'" + "'4.33.3'" ] }, - "execution_count": 13, + "execution_count": 12, "metadata": {}, "output_type": "execute_result" } @@ -240,47 +246,19 @@ }, { "cell_type": "code", - "execution_count": null, - "id": "16e06ee5-c4ca-4211-8e24-dbfd84b1d85b", - "metadata": {}, - "outputs": [], - "source": [ - "设置为国内可访问" - ] - }, - { - "cell_type": "code", - "execution_count": 1, - "id": "3e07ddc9-187e-4482-a7b5-7e4e9381d805", - "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "env: HF_ENDPOINT=https://hf-mirror.com\n" - ] - } - ], - "source": [ - "%env HF_ENDPOINT=https://hf-mirror.com" - ] - }, - { - "cell_type": "code", - "execution_count": 2, + "execution_count": 13, "id": "a9839137-b8ea-4a2c-85fc-9a63064cf8c8", "metadata": {}, "outputs": [ { "data": { "application/vnd.jupyter.widget-view+json": { - "model_id": "afc151b540664287aa60a4cbe90cdfeb", + "model_id": "ef488b57e4214b76a8913a4704de7e15", "version_major": 2, "version_minor": 0 }, "text/plain": [ - "vocab.json: 0.00B [00:00, ?B/s]" + "vocab.json: 0%| | 0.00/1.04M [00:00)" - ] - }, - "execution_count": 5, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "embedding(torch.tensor([1]))" - ] - }, - { - "cell_type": "markdown", - "id": "6a4d47f2-4691-47b8-9855-2528b6c285c9", - "metadata": {}, - "source": [ - "- Below is a visualization of what happens under the hood:" - ] - }, - { - "cell_type": "markdown", - "id": "12ffd155-7190-44b1-b6b6-45b11d6fe83b", - "metadata": {}, - "source": [ - "" - ] - }, - { - "cell_type": "markdown", - "id": "87d1311b-cfb2-4afc-9e25-e4ecf35370d9", - "metadata": {}, - "source": [ - "- Similarly, we can use embedding layers to obtain the vector representation of a training example with ID 2:" - ] - }, - { - "cell_type": "code", - "execution_count": 6, - "id": "c309266a-c601-4633-9404-2e10b1cdde8c", - "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "tensor([[ 0.6957, -1.8061, -1.1589, 0.3255, -0.6315]],\n", - " grad_fn=)" - ] - }, - "execution_count": 6, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "embedding(torch.tensor([2]))" - ] - }, - { - "cell_type": "markdown", - "id": "7ad3b601-f68c-41b1-a28d-b624b94ef383", - "metadata": {}, - "source": [ - "" - ] - }, - { - "cell_type": "markdown", - "id": "27dd54bd-85ae-4887-9c5e-3139da361cf4", - "metadata": {}, - "source": [ - "- Now, let's convert all the training examples we have defined previously:" - ] - }, - { - "cell_type": "code", - "execution_count": 7, - "id": "0191aa4b-f6a8-4b0d-9c36-65e82b81d071", - "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "tensor([[ 0.6957, -1.8061, -1.1589, 0.3255, -0.6315],\n", - " [-2.8400, -0.7849, -1.4096, -0.4076, 0.7953],\n", - " [ 1.3010, 1.2753, -0.2010, -0.1606, -0.4015]],\n", - " grad_fn=)" - ] - }, - "execution_count": 7, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "idx = torch.tensor([2, 3, 1])\n", - "embedding(idx)" - ] - }, - { - "cell_type": "markdown", - "id": "146cf8eb-c517-4cd4-aa91-0e818fed7651", - "metadata": {}, - "source": [ - "- Under the hood, it's still the same look-up concept:" - ] - }, - { - "cell_type": "markdown", - "id": "b392eb43-0bda-4821-b446-b8dcbee8ae00", - "metadata": {}, - "source": [ - "" - ] - }, - { - "cell_type": "markdown", - "id": "f0fe863b-d6a3-48f3-ace5-09ecd0eb7b59", - "metadata": {}, - "source": [ - "## Using nn.Linear" - ] - }, - { - "cell_type": "markdown", - "id": "138de6a4-2689-4c1f-96af-7899b2d82a4e", - "metadata": {}, - "source": [ - "- Now, we will demonstrate that the embedding layer above accomplishes exactly the same as `nn.Linear` layer on a one-hot encoded representation in PyTorch\n", - "- First, let's convert the token IDs into a one-hot representation:" - ] - }, - { - "cell_type": "code", - "execution_count": 8, - "id": "b5bb56cf-bc73-41ab-b107-91a43f77bdba", - "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "tensor([[0, 0, 1, 0],\n", - " [0, 0, 0, 1],\n", - " [0, 1, 0, 0]])" - ] - }, - "execution_count": 8, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "onehot = torch.nn.functional.one_hot(idx)\n", - "onehot" - ] - }, - { - "cell_type": "markdown", - "id": "aa45dfdf-fb26-4514-a176-75224f5f179b", - "metadata": {}, - "source": [ - "- Next, we initialize a `Linear` layer, which caries out a matrix multiplication $X W^\\top$:" - ] - }, - { - "cell_type": "code", - "execution_count": 9, - "id": "ae04c1ed-242e-4dd7-b8f7-4b7e4caae383", - "metadata": {}, - "outputs": [], - "source": [ - "torch.manual_seed(123)\n", - "linear = torch.nn.Linear(num_idx, out_dim, bias=False)" - ] - }, - { - "cell_type": "markdown", - "id": "63efb98e-5cc4-4e8d-9fe6-ef0ad29ae2d7", - "metadata": {}, - "source": [ - "- Note that the linear layer in PyTorch is also initialized with small random weights; to directly compare it to the `Embedding` layer above, we have to use the same small random weights, which is why we reassign them here:" - ] - }, - { - "cell_type": "code", - "execution_count": 10, - "id": "a3b90d69-761c-486e-bd19-b38a2988fe62", - "metadata": {}, - "outputs": [], - "source": [ - "linear.weight = torch.nn.Parameter(embedding.weight.T.detach())" - ] - }, - { - "cell_type": "markdown", - "id": "9116482d-f1f9-45e2-9bf3-7ef5e9003898", - "metadata": {}, - "source": [ - "- Now we can use the linear layer on the one-hot encoded representation of the inputs:" - ] - }, - { - "cell_type": "code", - "execution_count": 11, - "id": "90d2b0dd-9f1d-4c0f-bb16-1f6ce6b8ac2c", - "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "tensor([[ 0.6957, -1.8061, -1.1589, 0.3255, -0.6315],\n", - " [-2.8400, -0.7849, -1.4096, -0.4076, 0.7953],\n", - " [ 1.3010, 1.2753, -0.2010, -0.1606, -0.4015]], grad_fn=)" - ] - }, - "execution_count": 11, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "linear(onehot.float())" - ] - }, - { - "cell_type": "markdown", - "id": "f6204bc8-92e2-4546-9cda-574fe1360fa2", - "metadata": {}, - "source": [ - "As we can see, this is exactly the same as what we got when we used the embedding layer:" - ] - }, - { - "cell_type": "code", - "execution_count": 12, - "id": "2b057649-3176-4a54-b58c-fd8fbf818c61", - "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "tensor([[ 0.6957, -1.8061, -1.1589, 0.3255, -0.6315],\n", - " [-2.8400, -0.7849, -1.4096, -0.4076, 0.7953],\n", - " [ 1.3010, 1.2753, -0.2010, -0.1606, -0.4015]],\n", - " grad_fn=)" - ] - }, - "execution_count": 12, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "embedding(idx)" - ] - }, - { - "cell_type": "markdown", - "id": "0e447639-8952-460e-8c8f-cf9e23c368c9", - "metadata": {}, - "source": [ - "- What happens under the hood is the following computation for the first training example's token ID:" - ] - }, - { - "cell_type": "markdown", - "id": "1830eccf-a707-4753-a24a-9b103f55594a", - "metadata": {}, - "source": [ - "" - ] - }, - { - "cell_type": "markdown", - "id": "9ce5211a-14e6-46aa-a3a8-14609f086e97", - "metadata": {}, - "source": [ - "- And for the second training example's token ID:" - ] - }, - { - "cell_type": "markdown", - "id": "173f6026-a461-44da-b9c5-f571f8ec8bf3", - "metadata": {}, - "source": [ - "" - ] - }, - { - "cell_type": "markdown", - "id": "e2608049-f5d1-49a9-a14b-82695fc32e6a", - "metadata": {}, - "source": [ - "- Since all but one index in each one-hot encoded row are 0 (by design), this matrix multiplication is essentially the same as a look-up of the one-hot elements\n", - "- This use of the matrix multiplication on one-hot encodings is equivalent to the embedding layer look-up but can be inefficient if we work with large embedding matrices, because there are a lot of wasteful multiplications by zero" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "5eacc005-86fc-490c-8f6a-dc37d8a0df7c", - "metadata": {}, - "outputs": [], - "source": [] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "a1f63c81-1ee3-40a1-9ef2-14ff18fb4f05", - "metadata": {}, - "outputs": [], - "source": [] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "c71959bb-facf-44fd-8edb-b67f7752f034", - "metadata": {}, - "outputs": [], - "source": [] - } - ], - "metadata": { - "kernelspec": { - "display_name": "Python 3 (ipykernel)", - "language": "python", - "name": "python3" - }, - "language_info": { - "codemirror_mode": { - "name": "ipython", - "version": 3 - }, - "file_extension": ".py", - "mimetype": "text/x-python", - "name": "python", - "nbconvert_exporter": "python", - "pygments_lexer": "ipython3", - "version": "3.11.5" - } - }, - "nbformat": 4, - "nbformat_minor": 5 -} diff --git a/ch02/03_bonus_embedding-vs-matmul/README.md b/ch02/03_bonus_embedding-vs-matmul/README.md index a1f67ef..5228286 100644 --- a/ch02/03_bonus_embedding-vs-matmul/README.md +++ b/ch02/03_bonus_embedding-vs-matmul/README.md @@ -1,3 +1,4 @@ -# Chapter 2: Working with Text Data +# 第2章:使用文本数据 + +- [embeddings-and-linear-layers.ipynb](embeddings-and-linear-layers.ipynb) 包含可选(奖励)代码,以说明应用于独热编码向量的嵌入层和全连接层是等效的。 -- [embeddings-and-linear-layers.ipynb](embeddings-and-linear-layers.ipynb) contains optional (bonus) code to explain that embedding layers and fully connected layers applied to one-hot encoded vectors are equivalent. diff --git a/ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb b/ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb index 198e1b1..e068fc5 100644 --- a/ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb +++ b/ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb @@ -5,7 +5,7 @@ "id": "063850ab-22b0-4838-b53a-9bb11757d9d0", "metadata": {}, "source": [ - "# Embedding Layers and Linear Layers" + "# Embedding层和 Linear层" ] }, { @@ -13,8 +13,8 @@ "id": "0315c598-701f-46ff-8806-15813cad0e51", "metadata": {}, "source": [ - "- Embedding layers in PyTorch accomplish the same as linear layers that perform matrix multiplications; the reason we use embedding layers is computational efficiency\n", - "- We will take a look at this relationship step by step using code examples in PyTorch" + "- 在PyTorch中,嵌入层(Embedding layers)实现了执行矩阵乘法的线性层的相同功能;我们使用嵌入层的原因是为了提高计算效率。\n", + "- 我们将逐步使用PyTorch中的代码示例来查看这种关系。" ] }, { @@ -27,13 +27,12 @@ "name": "stdout", "output_type": "stream", "text": [ - "PyTorch version: 2.1.0.post301\n" + "PyTorch version: 1.12.1+cu113\n" ] } ], "source": [ "import torch\n", - "\n", "print(\"PyTorch version:\", torch.__version__)" ] }, @@ -47,22 +46,21 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": 3, "id": "cc489ea5-73db-40b9-959e-0d70cae25f40", "metadata": {}, "outputs": [], "source": [ - "# Suppose we have the following 3 training examples,\n", - "# which may represent token IDs in a LLM context\n", + "# 假设我们有以下 3 个训练样本,\n", + "# 这些样本可能表示语言模型(LM)上下文中的标记ID\n", "idx = torch.tensor([2, 3, 1])\n", "\n", - "# The number of rows in the embedding matrix can be determined\n", - "# by obtaining the largest token ID + 1.\n", - "# If the highest token ID is 3, then we want 4 rows, for the possible\n", - "# token IDs 0, 1, 2, 3\n", - "num_idx = max(idx)+1\n", + "# 嵌入矩阵的行数可以通过获取最大标记ID + 1 来确定。\n", + "# 如果最高的标记ID是3,则我们希望有4行,对应可能的\n", + "# 标记ID 0, 1, 2, 3\n", + "num_idx = max(idx) + 1\n", "\n", - "# The desired embedding dimension is a hyperparameter\n", + "# 所需的嵌入维度是一个超参数\n", "out_dim = 5" ] }, @@ -71,21 +69,21 @@ "id": "93d83a6e-8543-40af-b253-fe647640bf36", "metadata": {}, "source": [ - "- Let's implement a simple embedding layer:" + "- 实现一个简单的嵌入层" ] }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 4, "id": "60a7c104-36e1-4b28-bd02-a24a1099dc66", "metadata": {}, "outputs": [], "source": [ - "# We use the random seed for reproducibility since\n", - "# weights in the embedding layer are initialized with\n", - "# small random values\n", + "# 为了可重复性,我们使用随机种子,\n", + "# 因为嵌入层的权重是用小的随机值初始化的\n", "torch.manual_seed(123)\n", "\n", + "# 创建一个嵌入层,指定输入维度为 num_idx,输出维度为 out_dim\n", "embedding = torch.nn.Embedding(num_idx, out_dim)" ] }, @@ -94,12 +92,12 @@ "id": "dd96c00a-3297-4a50-8bfc-247aaea7e761", "metadata": {}, "source": [ - "We can optionally take a look at the embedding weights:" + "查看嵌入权重数据情况" ] }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 5, "id": "595f603e-8d2a-4171-8f94-eac8106b2e57", "metadata": {}, "outputs": [ @@ -113,7 +111,7 @@ " [-2.8400, -0.7849, -1.4096, -0.4076, 0.7953]], requires_grad=True)" ] }, - "execution_count": 4, + "execution_count": 5, "metadata": {}, "output_type": "execute_result" } @@ -127,12 +125,12 @@ "id": "c86eb562-61e2-4171-ab6e-b410a1fd5c18", "metadata": {}, "source": [ - "- We can then use the embedding layers to obtain the vector representation of a training example with ID 1:" + "- 使用嵌入层来获取具有ID 1的训练样本的向量表示" ] }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 6, "id": "8bbc0255-4805-4be9-9f4c-1d0d967ef9d5", "metadata": {}, "outputs": [ @@ -143,7 +141,7 @@ " grad_fn=)" ] }, - "execution_count": 5, + "execution_count": 6, "metadata": {}, "output_type": "execute_result" } @@ -157,7 +155,7 @@ "id": "6a4d47f2-4691-47b8-9855-2528b6c285c9", "metadata": {}, "source": [ - "- Below is a visualization of what happens under the hood:" + "- 下面是底层操作的可视化" ] }, { @@ -173,12 +171,12 @@ "id": "87d1311b-cfb2-4afc-9e25-e4ecf35370d9", "metadata": {}, "source": [ - "- Similarly, we can use embedding layers to obtain the vector representation of a training example with ID 2:" + "- 同样,我们可以使用嵌入层来获取具有ID 2的训练样本的向量表示:" ] }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 7, "id": "c309266a-c601-4633-9404-2e10b1cdde8c", "metadata": {}, "outputs": [ @@ -189,7 +187,7 @@ " grad_fn=)" ] }, - "execution_count": 6, + "execution_count": 7, "metadata": {}, "output_type": "execute_result" } @@ -211,12 +209,12 @@ "id": "27dd54bd-85ae-4887-9c5e-3139da361cf4", "metadata": {}, "source": [ - "- Now, let's convert all the training examples we have defined previously:" + "- 现在,让我们将之前定义的所有训练样本转换:" ] }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 8, "id": "0191aa4b-f6a8-4b0d-9c36-65e82b81d071", "metadata": {}, "outputs": [ @@ -229,12 +227,13 @@ " grad_fn=)" ] }, - "execution_count": 7, + "execution_count": 8, "metadata": {}, "output_type": "execute_result" } ], "source": [ + "# 将原先的第三行变成现在的第一行,第四行变成现在的第二行,第二行变成现在的第三行\n", "idx = torch.tensor([2, 3, 1])\n", "embedding(idx)" ] @@ -260,7 +259,7 @@ "id": "f0fe863b-d6a3-48f3-ace5-09ecd0eb7b59", "metadata": {}, "source": [ - "## Using nn.Linear" + "## 使用 nn.Linear" ] }, { @@ -268,13 +267,13 @@ "id": "138de6a4-2689-4c1f-96af-7899b2d82a4e", "metadata": {}, "source": [ - "- Now, we will demonstrate that the embedding layer above accomplishes exactly the same as `nn.Linear` layer on a one-hot encoded representation in PyTorch\n", - "- First, let's convert the token IDs into a one-hot representation:" + "- 接下来,我们将使用One-Hot编码,与embedding 层一样,在 `nn.Linear` 层进行操作\n", + "- 首先,我们将标记ID转换为One-Hot表示:" ] }, { "cell_type": "code", - "execution_count": 8, + "execution_count": 12, "id": "b5bb56cf-bc73-41ab-b107-91a43f77bdba", "metadata": {}, "outputs": [ @@ -286,7 +285,7 @@ " [0, 1, 0, 0]])" ] }, - "execution_count": 8, + "execution_count": 12, "metadata": {}, "output_type": "execute_result" } @@ -301,18 +300,33 @@ "id": "aa45dfdf-fb26-4514-a176-75224f5f179b", "metadata": {}, "source": [ - "- Next, we initialize a `Linear` layer, which caries out a matrix multiplication $X W^\\top$:" + "- 接下来,我们使用矩阵乘法$X W^\\top$ 来初始化一个Linear层" ] }, { "cell_type": "code", - "execution_count": 9, + "execution_count": 16, "id": "ae04c1ed-242e-4dd7-b8f7-4b7e4caae383", "metadata": {}, - "outputs": [], + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Parameter containing:\n", + "tensor([[-0.2039, 0.0166, -0.2483, 0.1886],\n", + " [-0.4260, 0.3665, -0.3634, -0.3975],\n", + " [-0.3159, 0.2264, -0.1847, 0.1871],\n", + " [-0.4244, -0.3034, -0.1836, -0.0983],\n", + " [-0.3814, 0.3274, -0.1179, 0.1605]], requires_grad=True)\n" + ] + } + ], "source": [ "torch.manual_seed(123)\n", - "linear = torch.nn.Linear(num_idx, out_dim, bias=False)" + "# 初始化一个Linear层,该层的权重矩阵是由 num_idx(输入维度)到 out_dim(输出维度)的一个线性层,而且没有偏置项\n", + "linear = torch.nn.Linear(num_idx, out_dim, bias=False)\n", + "print(linear.weight)" ] }, { @@ -320,16 +334,17 @@ "id": "63efb98e-5cc4-4e8d-9fe6-ef0ad29ae2d7", "metadata": {}, "source": [ - "- Note that the linear layer in PyTorch is also initialized with small random weights; to directly compare it to the `Embedding` layer above, we have to use the same small random weights, which is why we reassign them here:" + "- 请注意,PyTorch中的`linear`层也是用小的随机权重进行初始化的。为了与上面的 `Embedding` 层进行直接比较,我们必须使用相同的小随机权重,这就是我们在这里重新分配它们的原因:" ] }, { "cell_type": "code", - "execution_count": 10, + "execution_count": 17, "id": "a3b90d69-761c-486e-bd19-b38a2988fe62", "metadata": {}, "outputs": [], "source": [ + "# linear 层的权重就被重新赋值为与 embedding 层相同的小随机权重,以确保它们具有相同的初始化。这是为了使它们在后续操作中可以进行直接比较。\n", "linear.weight = torch.nn.Parameter(embedding.weight.T.detach())" ] }, @@ -338,12 +353,12 @@ "id": "9116482d-f1f9-45e2-9bf3-7ef5e9003898", "metadata": {}, "source": [ - "- Now we can use the linear layer on the one-hot encoded representation of the inputs:" + "- 现在,我们可以使用线性层处理输入的One-Hot编码表示:" ] }, { "cell_type": "code", - "execution_count": 11, + "execution_count": 18, "id": "90d2b0dd-9f1d-4c0f-bb16-1f6ce6b8ac2c", "metadata": {}, "outputs": [ @@ -355,7 +370,7 @@ " [ 1.3010, 1.2753, -0.2010, -0.1606, -0.4015]], grad_fn=)" ] }, - "execution_count": 11, + "execution_count": 18, "metadata": {}, "output_type": "execute_result" } @@ -369,12 +384,12 @@ "id": "f6204bc8-92e2-4546-9cda-574fe1360fa2", "metadata": {}, "source": [ - "As we can see, this is exactly the same as what we got when we used the embedding layer:" + "正如我们所看到的,这与我们使用嵌入层时得到的结果完全相同:" ] }, { "cell_type": "code", - "execution_count": 12, + "execution_count": 19, "id": "2b057649-3176-4a54-b58c-fd8fbf818c61", "metadata": {}, "outputs": [ @@ -387,7 +402,7 @@ " grad_fn=)" ] }, - "execution_count": 12, + "execution_count": 19, "metadata": {}, "output_type": "execute_result" } @@ -401,7 +416,7 @@ "id": "0e447639-8952-460e-8c8f-cf9e23c368c9", "metadata": {}, "source": [ - "- What happens under the hood is the following computation for the first training example's token ID:" + "- 底层发生的计算如下,针对第一个训练样本的标记ID:" ] }, { @@ -417,7 +432,7 @@ "id": "9ce5211a-14e6-46aa-a3a8-14609f086e97", "metadata": {}, "source": [ - "- And for the second training example's token ID:" + "- 以及对于第二个训练样本的标记ID:" ] }, { @@ -433,8 +448,9 @@ "id": "e2608049-f5d1-49a9-a14b-82695fc32e6a", "metadata": {}, "source": [ - "- Since all but one index in each one-hot encoded row are 0 (by design), this matrix multiplication is essentially the same as a look-up of the one-hot elements\n", - "- This use of the matrix multiplication on one-hot encodings is equivalent to the embedding layer look-up but can be inefficient if we work with large embedding matrices, because there are a lot of wasteful multiplications by zero" + "- \n", + "由于每个独热编码行中除了一个索引外都为0(设计如此),这个矩阵乘法本质上就是对独热编码元素的查找\n", + "- 。在独热编码上使用矩阵乘法与使用嵌入层查找是等效的,但如果我们使用大型嵌入矩阵,这种方法可能效率较低,因为有很多不必要的零乘法。" ] }, { @@ -444,22 +460,6 @@ "metadata": {}, "outputs": [], "source": [] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "a1f63c81-1ee3-40a1-9ef2-14ff18fb4f05", - "metadata": {}, - "outputs": [], - "source": [] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "c71959bb-facf-44fd-8edb-b67f7752f034", - "metadata": {}, - "outputs": [], - "source": [] } ], "metadata": { @@ -478,7 +478,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.11.5" + "version": "3.8.17" } }, "nbformat": 4,