How to set AdamW's weight decay as you scale model and dataset size | Read Paper on Bytez