add comments for megatron training