TensorFlow学習パラメータのsave, restoreでつまった - Qiita

TensorFlowのsave, restoreで少しつまったのでメモ． 基本的な使い方 Tensorflowの学習パラーメータのsave, restoreにはtf.train.Saverを用います． 保存 保存にはtf.train.Saverのsaveメソッドを使います． # 幾つか変数作成 w1 = tf.Variable(..., name="v1") w2 = tf.Variable(..., name="w2") v1 = tf.Variable(..., name="v1") v2 = tf.Variable(..., name="v2") ... saver = tf.train.Saver() ... # 保存 saver.save(sess, "model.ckpt", global_step=100) 上記により，変数w1,w2,b1,b2がmodel.ckpkに保存さ