search

layumi / Person_reID_baseline_pytorch

:bouncing_ball_person: Pytorch ReID: A tiny, friendly, strong pytorch implement of person re-id / vehicle re-id baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial
https://www.zdzheng.xyz
MIT License
4.06k stars 1k forks source link

关于torch2trt推理加速问题 #323

Open NUISTGY opened 2 years ago

NUISTGY commented 2 years ago

test_with_TensorRT.py文件中第294行的 model_trt = torch2trt(model, [x]) 实现了模型的转换，但是后续代码中并没有再出现对model_trt的调用，本质上还是使用了未加速的模型。请问应如何更改呢？如是我的理解有误，烦请作者指出，万分感谢！

layumi commented 2 years ago

感谢 ！ 代码已更新。 目前 可以把test 从 40s 缩短到 23s。