유용한 코드 스크랩

learning rate active하게 조정 (https://lcyking.tistory.com/78)

from tensorflow.keras.callbacks import ReduceLROnPlateau, EarlyStopping

# 3번 반복내에 validation loss가 줄어들지 않으면 learning rate를 0.2 감소
lr_cb = ReduceLROnPlateau(monitor='val_loss', factor=0.2, patience=3, mode='min', verbose=1)
# 5번 반복내에 validation loss가 줄어들지 않으면 강제종료
st_cb = EarlyStopping(monitor='val_loss', patience=5, mode='min', verbose=1)

 

전이학습 가중치 불러오기 (https://velog.io/@agzg/Tensorflow%EC%A0%84%EC%9D%B4%ED%95%99%EC%8A%B5Transfer-Learning)

import os
from tensorflow.keras import layers
from tensorflow.keras import Model
from tensorflow.keras.applications.inception_v3 import InceptionV3

local_weights_file = '/tmp/inception_v3_weights_tf_dim_ordering_tf_kernels_notop.h5'

pre_trained_model = InceptionV3(input_shape=(150, 150, 3),
                              include_top=False,
                              weights=None)

pre_trained_model.load_weights(local_weights_file)

for layer in pre_trained_model.layers:
  layer.trainable = False

# Model save_weights
model.save_weights('./checkpoints/my_only_checkpoint')