Auto pick num nodes in preparing megatron ckpt#607

fzyzcjy wants to merge 3 commits intofeat/verify_doneradixark/miles:feat/verify_donefrom

feat/generalize_num_nodesradixark/miles:feat/generalize_num_nodesCopy head branch name to clipboard