Transformer Model In Pytorch