osmarks/nanogpt-experiments
1
0
Fork 0
mirror of https://github.com/osmarks/nanogpt-experiments.git synced 2024-12-18 14:10:28 +00:00
Code Issues Releases Wiki Activity

bugfix in decode function

Browse Source 
Return was left out of the decoder, so it didn't work.
This commit is contained in:
Christian Orr 2023-03-08 10:16:19 +02:00 committed by GitHub
parent 0d8fbd11ae
commit 36c7db8c44
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/shakespeare_char/prepare.py
View File

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