Skip to content

Commit 57a16a0

Browse files
authored
[Fix] Fix the export shape of rtformer in tipc
1 parent 8fc5fbd commit 57a16a0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test_tipc/configs/rtformer/train_infer_python.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ export1:null
3535
export2:null
3636
===========================infer_params===========================
3737
infer_model:./test_tipc/output/rtformer_base/model.pdparams
38-
infer_export:export.py --config test_tipc/configs/rtformer/rtformer_base_cityscapes_1024x512_120k.yml
38+
infer_export:export.py --config test_tipc/configs/rtformer/rtformer_base_cityscapes_1024x512_120k.yml --input_shape 1 3 512 512
3939
infer_quant:False
4040
inference:deploy/python/infer.py --save_dir ./test_tipc/output/rtformer_base/
4141
--device:cpu|gpu

0 commit comments

Comments
 (0)