1
0
mirror of https://github.com/osmarks/nanogpt-experiments.git synced 2025-08-30 17:37:58 +00:00

Merge pull request #199 from ChristianOrr/patch-1

bugfix in decode function
This commit is contained in:
Andrej
2023-03-12 13:40:20 -07:00
committed by GitHub

View File

@@ -32,7 +32,7 @@ itos = { i:ch for i,ch in enumerate(chars) }
def encode(s):
return [stoi[c] for c in s] # encoder: take a string, output a list of integers
def decode(l):
''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string
return ''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string
# create the train and test splits
n = len(data)