GitLab

* cirtical performance fix - reduce atomic

 * training faster up to 50%

* leave a TODO
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>