[TensorFlow] Model Save/Load 관련

TF Model Save

1. tf.saved_model.save(model, path) — Keras
   • Output
     • path/saved_model.pb
     • path/assets
     • path/variables
       • variables.data-00000-of-00001
       • variables.index
2. model.save_weights(path)
   • path 내 variables (Weights) 만 변경하는 방식
   • Output
     • variables.data
     • variables.index

TF Model Load

1. tf.saved_model.load(path) — Keras
   • pb 파일 불러올 때 쓴다
2. model.load_weights(path)
   • model 의 weight 값을 불러올 때 쓴다

variables.data → saved_model.pb

1. model.load_weights(variables_path)
2. model.save(output_path)