Model Load (1) 썸네일형 리스트형 [Pytorch] 모델 저장하고 불러오기 파이토치 한국사용자 모임의 글을 보면서 공부한 내용을 정리한 글입니다. 모델을 저장하고 불러오기를 통해 모델의 상태를 유지할 수 있다. 모델 가중치 저장 Pytorch 모델은 학습한 매개변수를 state_dict라고 불리는 내부 상태 사전(Internal State Dictionary)에 저장한다. 이 상태 값들은 torch.save 메소드를 사용하여 저장할 수 있다. import torch import torchvision.models as models model = models.vgg16(weights='IMAGENET1K_V1') # torch.save(모델의 상태 사전, 저장할 경로) torch.save(model.state_dict(), 'model_weights.pth') 모델 가중치 불러오기.. 이전 1 다음
