[RFC] Add lr into metric logging and also rename the loss name#939

fduwjj wants to merge 2 commits intomainpytorch/torchtitan:mainfrom

tb_logspytorch/torchtitan:tb_logsCopy head branch name to clipboard