Logs more detailed mse info now.
Added flag for running a maximum number of epochs.