Describe the bug

1. What command or script did you run?
   python deploy.py ocrnet18/deploy/segmentation_tensorrt-fp16_static-512x512.py ocrnet18/ocrnet_hr18s_512x1024_160k_cityscapes.py mmsegmentation-0.21.1/tools/work_dirs/ocrnet_hr18s_512x1024_160k_cityscapes/iter_160000.pth /mnt/net_school/libu_test_dataset/test.jpg --test-img /mnt/net_school/cityscapes/leftImg8bit/val/ --dump-info --work-dir /home/zj/pass_seg/ocrnet18/

2. I found the code is wrong, How to fix it?