Fix arguments.
Fix Nesterov's momentum.