momentum and dropout added.
training options added