Set shuffle to true by default.
fix momentum in mlp