Add learning rate and momentum to config file