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))