shakespeareGPT
1.0.0
基于Andrej Karpathy的建筑和培训GPT从头开始:让我们构建GPT:从头开始,代码,阐明。教程
使用字符级令牌。训练有两个具有不同配置的版本,以更好地了解超参数,例如N_Embeds,num_heads。
V1:
v2:
使用了字节对编码令牌。
huggingface tokenizers从头开始的BPE令牌,类似于保存在Tokenizer上的GPT-2 V1
n_embed = 384
n_heads = 12
head_size = 32
n_layers = 4
lr = 6e-4
attn_dropout = 0.1
block_dropout = 0.1
Train Loss: 4.020419597625732
Valid Loss: 6.213085174560547
V2
n_embed = 384
n_heads = 6
head_size = 64
n_layers = 3
lr = 5e-4
attn_dropout = 0.2
block_dropout = 0.2
Train Loss: 3.933095216751099
Valid Loss: 5.970513820648193
和往常一样,安德烈·伊斯特里(Andrej)的一个令人难以置信的教程!