step.yaml 84 Bytes
Newer Older
Tri Dao's avatar
Tri Dao committed
1
2
3
# @package train.scheduler
_target_: torch.optim.lr_scheduler.StepLR
step_size: ???