목록pytorch (1)

AI 탐구노트

AI 모델의 모델 가중치 파일 형식

딥러닝 모델의 가중치 파일은 프레임워크나 용도에 맞춰 최적화되어 있습니다. 주요 가중치 파일의 형식과 특징을 정리해 보고 가겠습니다.   HDF5 (.h5, .hdf5)Hierachical Data Format의 약자, 대용량 데이터 저장에 적합모델의 구조와 가중치를 동시에 저장Keras, Tensorflow에서 사용 PyTorch (.pt, .pth)Pytorch에서 사용모델의 상태(dict 형태)를 저장가중치와 옵티마이저 상태 정보를 포함할 수 있음 TensorFlow SavedModel:Tensorflow에서 사용모델의 가중치, 구조, 서명, 메타데이터 포함디렉토리 구조로 저장되며 saved_model.pb 파일과 가중치 파일들이 함께 저장됨Tensorflow Serving, Tensorflow L..
기술 팁 2024. 10. 7. 12:14
이전 Prev 1 Next 다음