Implementation and necessary adaption for loading our dataset and training on it. Including custom model creation and validation.