딥러닝모델 or 가중치(wieght) 저장하는 법

김우빈·2022년 6월 1일
0

JUST DEEP_HINT

목록 보기
1/6
post-thumbnail

모델 저장 코드

#전체모델 학습된 모델을 (가중치 포함)h5형태로 저장
model.save("./YOUR_MODEL_NAME.h5")
print("save succeed")#확인

가중치(weight)가 포함된 모델입니다.

가중치(Weight) 저장 코드

model.save_weights('./YOUR_MODELS_WEIGHT_NAME.h5')
print("save succeed")

오늘도 깊은하루 되세요^^

profile
DeepLearning, MLOps

0개의 댓글