Notebook Solution: https://colab.research.google.com/gist/Jeremy26/13f71c273f0a1a93f758d02b2b77802e/segformer-cityscapes-run.ipynb?authuser=0#scrollTo=54M7rPQBQYA1

Next, let's see how the Decoder works. You can try now, and then try again after watching the video if you didn't get how the decoder part works.