파이토치 모델 로딩, pytorch model loading 파이토치로 학습시킨 모델의 파라미터들을 파일로 저장해 두었다면, 다시 그 파일을 모델에 읽어들여서 사용할 필요가 있습니다. 여기서는 파일로부터 모델 파라미터들을 읽어서 모델에 로딩시키는 과정을 설명합니다. 1단계 파일로 저장된 모델을 dictionary로 읽어들여야 한다. 파일 확장자는 .pth, .pt를 사용합니다. 모델 읽기를 위해서 파이토치에서는 torch.load( )라는 함수를 제공합니다. _state_dict = torch.load('saved_model.pth') _state_dict는 Python dictionary type으로, keyd와 value 쌍으로 이루어져 있습니다. 2단계 Dictionary에 읽혀진 파라미터들의 이름을 바꾸어야 할 때가 있습니다. 아래 설명한 경우가 아니라면,.. 더보기 이전 1 2 3 4 5 6 7 ··· 11 다음
