파이토치 모델 Save & Load


파이토치 모델 Save & Load

Save Load 직렬화 역직렬화 Torch 모델 사용법 파이토치에서는 모델을 구성하고나서 훈련을 진행합니다. 하지만 모델을 새롭게 사용할 때마다 훈련을 진행하기에는 너무나 많은 자원이 소모됩니다. 따라서 저희는 모델의 파라미터를 저장 & 불러오기 또는 모델 자체를 저장 & 불러오기 할 수 있습니다. 이러한 개념에 대해서 이해하기 위해 우선 직렬화, 역직렬화에 대해 알아봅니다. 직렬화 & 역직렬화 간단합니다. 기존에 어떤 형태를 가지는 데이터들은 전부 메모리에 공간을 차지하면서 형태를 유지하게 되는데 우리는 이러한 메모리에 존재하는 값까지 모델을 옮길 때 이동시키고 싶습니다. 이 때 필요한 방법이 특정 파일에 모든 데이터에 대한 내용을 넣어서 옮기는 것인데 이를 직렬화라고 합니다. 직렬화에는 몇 가지 방법이 존재합니다. CSV, XML, JSON 직렬화 DL에서 데이터를 다룰 때 많이 볼 수 있는 방법입니다. 파일에 어떤 데이터들의 모든 메타 정보가 포함되어 있어서, 간단하게 사용할...


#CheckPoint #Torch #모델 #모델FineTune #모델로드 #모델만저장하는법 #모델저장 #모델파라미터 #역직렬화 #직렬화 #파이토치모델 #state_dict #save #Pytorch #CPU #CudaModel #GPU #Load #load_state_dict #map_location #model_state_dict #parameters #params #Pickle #파이토치모델로딩

원문링크 : 파이토치 모델 Save & Load