# torch2trt

총 2개의 포스트
post-thumbnail

yolo v7 export.py 분석하기

어제 torch.jit.script 활용하여 pt 파일을 생성해보려 했지만, 제대로 되지 않았다.따라서, yolo v7의 소스코드에 있는 export.py 를 분석하여, 어떤 방식으로 pyTorch 모델을 architecture와 parameter가 담긴 pt파일이

OnnxTensorRTinferencetorch.jit.scripttorch.jit.tracetorch2onnxtorch2trtyolov7
2022년 9월 27일
·
0개의 댓글
·
post-thumbnail

torch2trt 설치 중 오류 해결

https://github.com/NVIDIA-AI-IOT/torch2trt설치 중 Step 2. 에서 아래의 오류가 발생하는 경우가 있다.tensorrt 헤더의 경로가 추가되지 않은 경우이며,https://github.com/NVIDIA-AI-IOT

TensorRTtorch2trt
2022년 6월 9일
·
0개의 댓글
·