[references] Unify sched + optim config and add AdamW as option (#1825)

[references] Unify sched + optim config and add AdamW as option (#1825) #3379