[Deep Learning] TensorFlow 모델 저장, 불러오기

1. 모델을 폴더로 저장, 불러오기

>>> model.save('my_model')

# 불러오기
>>> my_model = tf.keras.models.load_model('my_model')

 

2. 모델을 파일로 저장, 불러오기

>>> model.save('my_model.h5')

# 불러오기
>>> my_model2 = tf.keras.models.load_model('my_model.h5')

 

3. 웨이트 저장, 불러오기

>>> model.save_weights('my_model_weights.h5')

>>> model.load_weights('my_model_weight.h5')