Added the possibility to have different learning rates per layer.