add support for diifferent learning rate.