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

4 lines
84 B
YAML

# @package train.scheduler
_target_: torch.optim.lr_scheduler.StepLR
step_size: ???