# torch.jit.trace

2개의 포스트
post-thumbnail

Torch 커스텀 모델 만들어서 Test하기

if, else와 같이 분기를 갖는 코드의 경우, 어떤식으로 저장되는지 알아보기 위해, 간단하게 테스팅해보았다.우선, 지난번에 알아보았던, pyTorch의 Trace방식과 Script 방식으로 둘 다 저장해보고 시각화해서 진행해보려고한다.테스팅을 위해, 임의로 모델을

2022년 9월 27일
·
0개의 댓글
·
post-thumbnail

yolo v7 export.py 분석하기

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

2022년 9월 27일
·
0개의 댓글
·