홈 탐색 도움말
가입하기 로그인
david
/
flash-attention
의 미러 https://github.com/Dao-AILab/flash-attention
1
0
포크 0
파일 이슈 0 위키
트리: 0dfb281743
브랜치 태그
flash-attention
/
training
/
configs
/
model
/
gpt2model
/
gpt2-medium.yaml

gpt2-medium.yaml 85 B
히스토리 Raw

123456 
  1. # @package _global_
    2. 

  2. model:
    3. 

  3.   config:
    4. 

  4.     n_embd: 1024
    5. 

  5.     n_head: 16
    6. 

  6.     n_layer: 24
    7.