shakespeareGPT
1.0.0
Andrej Karpathyに基づいたゼロからGPTの構築とトレーニング:GPT:ゼロから、コードで、綴りをしましょう。チュートリアル
キャラクターレベルのトークンザーを使用しました。 n_embeds、num_headなどのハイパーパラメーターの影響をよりよく理解するために、異なる構成を備えた2つのバージョンをトレーニングしました。
V1:
V2:
トークン剤をエンコードするバイトペアを使用しました。
huggingface tokenizersで保存された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による信じられないほどのチュートリアル!