# torch2onnx

총 1개의 포스트
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개의 댓글
·