flash-attention/training/configs/trainer/default.yaml
2022-11-28 17:34:40 -08:00

8 lines
142 B
YAML

_target_: pytorch_lightning.Trainer
# set `gpu` to train on GPU, null to train on CPU only
accelerator: null
min_epochs: 1
max_epochs: 1000