TensorFlow は、変数名を Tensor 値にマップするバイナリ チェックポイント ファイル内の変数を保存します。vai_p_tensorflow は、チェックポイント ファイルを入力として取得し、学習済みの重みをロードします。tf.train.Saver
は、書き込まれるか、または読み出されるチェックポイント ファイルのパスを指定するメソッドです。
次のコードは、tf.train.Saver.save
メソッドを呼び出してチェックポイント ファイルに変数を保存します。
with tf.Session() as sess:
# your graph building codes here
# ……
sess.run(train_op)
# Save the variables to disk.
save_path = saver.save(sess, "/tmp/model.ckpt")
print("Model saved in path: %s" % save_path)
保存されたチェックポイント ファイルは次のようになります。
model.ckpt.data-00000-of-00001
model.ckpt.index
model.ckpt.meta