Used a higher batch size and a custom model for the training in this one. Dataset is largely the same though.

Really just an experiment with using different models outside of the base to see what the end results get me.