Early stopping

[AleksCipri]

Change early stopping to monitor validation loss, wince this is best for preventing overfitting.

[dkafkes]

updated to monitor validation loss, but need to figure out why it's stopping before plateau and focus now on fine-tuning learning rate parameters