Tensorflow 2.0 모델 저장, 불러오기

 Tensorflow 2.x에서 모델을 저장하고 불러오기 위한 코드입니다.

*.pb 파일을 불러오는 예제입니다.

import tensorflow as tf

pb_path = 'PATH TO DIR CONTAINING PB FILE'
model = tf.saved_model.load(pb_path)

tf.saved_model.save(model, "PATH TO SAVE MODEL")