problem of visualize

[Gatswei]

Great work! I met a problem when I try to visualize. How can I solve this problem?
python demo/image_demo.py demo/demo.png local_configs/segformer/B1/segformer.b1.512x512.ade.160k.py /data/segformer.b1.512x512.ade.160k.pth --device cuda:0 --palette cityscapes
Use load_from_local loader
Traceback (most recent call last):
File "demo/image_demo.py", line 29, in
main()
File "demo/image_demo.py", line 25, in main
show_result_pyplot(model, args.img, result, get_palette(args.palette))
File "/code/SegFormer-master/mmseg/apis/inference.py", line 115, in show_result_pyplot
img = model.show_result(img, result, palette=palette, show=False)
File "/code/SegFormer-master/mmseg/models/segmentors/base.py", line 246, in show_result
assert palette.shape[0] == len(self.CLASSES)
AssertionError