gpt3xl-flash-rotary-60B.yaml 162 Bytes
Newer Older
Tri Dao's avatar
Tri Dao committed
1
2
3
4
5
6
7
8
9
10
# @package _global_
defaults:
  - /experiment/pile/gpt2xl-flash-rotary.yaml

trainer:
  max_steps: 60000

train:
  scheduler:
    t_initial: ${trainer.max_steps}