flash-attention/training/configs/experiment/pile/gpt3m-hf.yaml
2022-11-29 04:46:43 -08:00

12 lines
173 B
YAML

# @package _global_
defaults:
- /experiment/pile/gpt3s-hf.yaml
- override /model/gpt2model: gpt2-medium
datamodule:
batch_size: 4
train:
optimizer:
lr: 3.0e-4