Corrections of 2 hidden layers; Learning rate schedule