Make rate an argument to Train, not a property of Layer.