Update legacy AdamW impl so it has a multi-tensor impl like NAdamW (n…#2517

rwightman merged 2 commits intomainhuggingface/pytorch-image-models:mainfrom

legacy_adadamw_updatehuggingface/pytorch-image-models:legacy_adadamw_updateCopy head branch name to clipboard