Fix MIMO optimizer setup for frozen modules#4790

liding-nv wants to merge 1 commit into

NVIDIA:mainNVIDIA/Megatron-LM:mainfrom

liding-nv:mimo-frozen-modulesliding-nv/Megatron-LM:mimo-frozen-modulesCopy head branch name to clipboard