Change sparse_init to use normal distribution instead of uniform
Implement StepDownMomentum