딥러닝/ 에포크 시마다 가장 좋은 모델을 저장하는 ModelCheckpoint

 

에포크 시마다 가장 좋은 모델을 저장하는 ModelCheckpoint

 

# 파일썬으로 디렉토리 만드는 코드 작성
if not os.path.exists(PROJECT_PATH + '/checkpoints/' + model_type + '/') :
  os.makedirs(PROJECT_PATH + '/checkpoints/' + model_type + '/')

 

from tensorflow.keras.callbacks import ModelCheckpoint

 

# 경로를 미리 변수로 만듬
CHECKPOINT_PATH
>> ./checkpoints/mobilenetv2/by-type-mobilenetv2-block-1-1.h5

 

cp = ModelCheckpoint(CHECKPOINT_PATH, monitor='val_accuracy', save_best_only=True, verbose=1)

 

힐링아무의 코딩일기 힐코딩!!