C++, Pytorch 적용하기 02 (.pt 만들기) - Visual Studio 2019

Python 환경에서 만든 모델을 libtorch에 적용하기 위해서는 다른 방식으로 .pt를 만들어 주어야 합니다.

 

총 두가지 방법이 있고 Pytorch 공식문서를 참고했습니다.

 

# torch.jit.trace 사용
model.eval()
example = torch.rand(1, 3, 64, 64)
trace_module = torch.jit.trace(model, example)
trace_module.save("./trace_module.pt")

# torch.jit.script 사용
script_module = torch.jit.script(model)
script_module.save("./script_module.pt")

 

위와 같은 방법으로 .pt 파일은 생성해주어야 libtorch에 적용 가능합니다.

 

일반적으로 .save 를 통해서 저장할 경우, 에러가 발생하게 됩니다.