* fix esmm

* fix esmm
Co-authored-by: Nwuzhihua <35824027+fuyinno4@users.noreply.github.com>
Co-authored-by: Ntangwei12 <tangwei12@baidu.com>