Remove print statements

Changes to the transformer layer

Add the dataset and the training script

Add the model and scheduler