"configs/vig/pvig-tiny_8xb128_in1k.py" did not exist on "62b8498e533f10bb78df9913691ec5322abd63e9"
gpt3l-hf.yaml 196 Bytes
Newer Older
Tri Dao's avatar
Tri Dao committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# @package _global_
defaults:
  - /experiment/pile/gpt3s-hf.yaml

model:
  config:
    n_embd: 1536
    n_head: 16
    n_layer: 24

datamodule:
  batch_size: 2

train:
  optimizer:
    lr: 2.5e-4