explicitly adding momentum term to the shared args dictionary