Finished the final fixes to the training parameters, Now the next part is training the module for hours on end on a dataset we get from kaggle!