Skip to content

SegFormer code

Compare
Choose a tag to compare
@poppuppy poppuppy released this 14 Jan 07:28
· 4 commits to main since this release
96a85cc

Please follow the installing, training and evaluation instructions at https://github.com/NVlabs/SegFormer.

For example, to train a SegFormer-B1 model with the DySample-lpg4 upsampler with a single 3090 GPU, one can run:

python tools/train.py local_configs/segformer/B1/segformer.b1_dysample-lpg4.512x512.ade.160k.py