모델불러오기1 [파이토치] 모델 저장 및 불러오기 이번 포스트에서는 학습한 모델을 저장하는 방법과 저장된 모델을 불러오는 방법에 대해 다뤄보려고 한다. 일반적으로 모델을 학습하는데 많은 시간이 걸린다. 그렇기 때문에 코드 파일을 매번 실행하는 것은 굉장히 비효율적이다. 그렇기 때문에 학습이 끝난 모델을 저장하고 나중에 사용할 때 그 모델을 불러와서 사용하는 것이 시간과 자원 면에서 효율성을 증대할 수 있다. 모델 저장과 불러오기 먼저 실습을 위해 필요한 기본적인 패키지를 import 한다. import torch import torchvision.models as models 모델을 저장하고 불러오는 방법은 크게 두 가지로 나뉜다. 모델의 가중치를 저장하고 불러오는 방식 저장하기 모델이 학습한 파라미터들은 state_dict에 저장된다. 그리고 이 st.. 2022. 10. 7. 이전 1 다음
