DeepSpeed

c39a76fe - Fix printing momentum for non-deepspeed optimizer (#464)

Fix printing momentum for non-deepspeed optimizer (#464) * Fix printing momentum for non-deepspeed optimizer Fix printing momentum for non-deepspeed optimizer * fix momentum access for Adam fix momentum access for Adam