ckpt = tf.train.get_checkpoint_state(FLAGS.model)
if ckpt:
saver.restore(sess, tf.train.latest_checkpoint(FLAGS.model))
else:
sess.run(init)
我原來的代碼是這樣的,會有什么問題,就是我從tfrecord讀數據的時候報一些參數沒有被初始化,
看來save的時候沒有把read tfrecord 的參數save,所以直接restore沒有run init會有問題,不能正常讀數據
我改成
sess.run(init)
ckpt = tf.train.get_checkpoint_state(FLAGS.model)
if ckpt:
saver.restore(sess, tf.train.latest_checkpoint(FLAGS.model))
