19 lines
408 B
YAML
19 lines
408 B
YAML
# @package _global_
|
|
defaults:
|
|
- /experiment/pile/gpt3xl-flash-rotary.yaml
|
|
|
|
model:
|
|
config:
|
|
n_embd: 2560
|
|
n_head: 20
|
|
n_layer: 32
|
|
initializer_range: ${eval:"(2 / (${.n_embd} * 5)) ** 0.5"}
|
|
mlp_checkpoint_lvl: 0
|
|
|
|
datamodule:
|
|
batch_size: ${eval:"4 if ${train.gpu_mem} < 24 else (8 if ${train.gpu_mem} < 40 else (16 if ${train.gpu_mem} < 80 else 32))"}
|
|
|
|
train:
|
|
optimizer:
|
|
lr: 1.6e-4
|