悪魔は優しい

なんでもやるし、なんにもできないIT屋メモ。

tensorflowで学習結果を保存するのはtf.train.Saver

calendar

reload

tensorflowで学習結果を保存するのはtf.train.Saver

保存するには

saver = tf.train.Saver()
saver.save(sess, "model.ckpt")

学習の途中でも保存は可能。

読み出すには

sess = tf.InteractiveSession()
saver = tf.train.Saver()
sess.run(tf.global_variables_initializer())
saver.restore(sess,FLAGS.readmodels)

この記事をシェアする

コメント

コメントはありません。

down コメントを残す