Add the training script for dit

Update to 10 epochs

Add the dataset and the training script