invsqrt.yaml 90 Bytes
Newer Older
Tri Dao's avatar
Tri Dao committed
1
2
3
# @package train.scheduler
_target_: src.optim.lr_scheduler.InvSqrt
num_warmup_steps: ???