Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
gpt2-medium.yaml 85 Bytes
Newer Older
Tri Dao's avatar
Release training code  
Tri Dao committed
1
2
3
4
5
6 
# @package _global_
model:
  config:
    n_embd: 1024
    n_head: 16
    n_layer: 24