add momentum and dropout