fix of adam optimizer as pointed in flashlight#143

Summary: fix for flashlight#143

Reviewed By: syhw, jacobkahn

Differential Revision: D22757150

fbshipit-source-id: 1c9267a7a594f7d3b7041389b7e40715cb14e6b4

readthe docs integration 2