# torch.jit.script
지난 이야기
기존에 계속해서 공부했던 내용을 정리하는 회고록입니다...기존에 If문 및 loop가 변환이 안 되는 것이 포워드 함수 내부의 파라미터가 Tensor인지 아닌지에 따라서 변환이 되고, 안되는 것이 달라진다고 생각했었습니다.위의 가설은 반은 맞고, 반은 틀린 내용이었습니
Custom Model torch.jit.script Debugging
커스텀 모델은 아래와 같이 만들었습니다.분기에 따라, (커스텀 레이어) → layer1 → layer2 → flatten → fully connected 커스텀 레이어 적용 시, If문이 포워드에서 사용됩니다.모델을 Torch Script로 실행하는 구문은 아래와 같습
custom model, torch.jit.script 따라가기
간단한 모델을 만들어서, torch.\_C.script에서 torch.jit.script 타입으로 변환되는 것인지, 아니면 torch.onnx 에서 변환되는 것인지 확인해보려고 합니다. torch.jit.script 공식 문서를 확인해보면, torch.\_C.scrip
torch script 활용하여 yolo v7 디버깅
yolov7 을 torch.jit.script로 export하는 과정을 debugging 해보려고 합니다.구조는 아래와 같습니다.기본적으로 torch.jit.script 로 export 할 때, 기준이 되는 부분이 models 디렉터리 안에, yolo.py 라고 생각을
torch.jit.script if,else 파악하기
지난번에 알아보았던, torch.jit.script 로 pt 파일을 내보낼 때, torch.eq 모듈이 지원되지 않는다며, 정상적으로 netron에서 열리지 않는 문제와 Onnx로 변환하였을 때, if,else와 같은 분기가 존재할 경우, Condition에 해당되는
Torch 커스텀 모델 만들어서 Test하기
if, else와 같이 분기를 갖는 코드의 경우, 어떤식으로 저장되는지 알아보기 위해, 간단하게 테스팅해보았다.우선, 지난번에 알아보았던, pyTorch의 Trace방식과 Script 방식으로 둘 다 저장해보고 시각화해서 진행해보려고한다.테스팅을 위해, 임의로 모델을
yolo v7 export.py 분석하기
어제 torch.jit.script 활용하여 pt 파일을 생성해보려 했지만, 제대로 되지 않았다.따라서, yolo v7의 소스코드에 있는 export.py 를 분석하여, 어떤 방식으로 pyTorch 모델을 architecture와 parameter가 담긴 pt파일이