: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
4.06k
stars
1k
forks
source link
关于torch2trt推理加速问题 #323
Open
NUISTGY opened 2 years ago
test_with_TensorRT.py文件中第294行的
model_trt = torch2trt(model, [x])
实现了模型的转换,但是后续代码中并没有再出现对model_trt的调用,本质上还是使用了未加速的模型。请问应如何更改呢?如是我的理解有误,烦请作者指出,万分感谢!