How to use:
python3 trace_segformer.py --config models/configs/segformer/segformer_config_b0.json --nclasses 8 --image_size 512 --weights /opt/platform/models/pretrained/torch/segformer/orig/segformer_mit-b0_512x512_160k_ade20k_20220617_162207-c00b9603.pth -o /opt/platform/models/pretrained/torch/segformer/