Shouldn't we get rid of the self.momentum and self.init_momentum fields altogether? We could just detect if someone uses those arguments here and convert them into a Momentum LearningRule.