This project runs with pytorch 1.11.0.

To run the code, use

python TransformerBatch.py