Added learning rate reset for pre-training after each layer, so hopefully the upper layers will not be underfit.
Also lowered the learning rate for finetuning and decreased the batch size.