Also change initial learning rate