Remove print statements

Remove print statements

Lesser print statements

Changes to the transformer layer

Made all changes needed to the patch embed layer for dimension mismatch

Add hidden_dim to patch embeddings

Changes in the patch_embed

Add the dataset and the training script

Add the lpip model

Add the model and scheduler