flash-attention/training/configs/optimizer/fusedlamb-ds.yaml
2022-11-28 17:34:40 -08:00

3 lines
66 B
YAML

# @package train.optimizer
_target_: deepspeed.ops.lamb.FusedLamb