fix: [mcore] only take optimizer steps when in train mode#1012

terrykong merged 1 commit intomainNVIDIA-NeMo/RL:mainfrom

ashors/megatron-bug-fixesNVIDIA-NeMo/RL:ashors/megatron-bug-fixesCopy head branch name to clipboard