Experimenting with different momentum and removed gradient arrays