model_stages_trt.py问题

[xinguta001]

当我调用model_stages_trt.py运行的时候，存在size不匹配问题，输入为（1，3，512，512），是否是我未进行正确的操作。