Bibliography (5):

1. Vision Transformer: An Image is Worth 16×16 Words: Transformers for Image Recognition at Scale

2. Attention Is All You Need

3. ImageNet: A Large-Scale Hierarchical Image Database

4. https://paperswithcode.com/dataset/ade20k

5. Wikipedia Bibliography:

   1. Loss function