"""
Modified from https://github.com/NVlabs/SegFormer
"""