TensorFlow | ValueError: Can't load save_path when it is None

原因:saver.restore(sess, tf.train.latest_checkpoint(’/tmp’)),找不到checkpoint所在路径,在restore之前添加判断checkpoint是否存在,若不存在则初始化的判断。

#判断模型保存路径是否存在,不存在就创建

if not os.path.exists('tmp/'):
    os.mkdir('tmp/')

#初始化

with tf.Session() as sess:
    if os.path.exists('tmp/checkpoint'):         #判断模型是否存在
        saver.restore(sess, 'tmp/model.ckpt')    #存在就从模型中恢复变量
    else:
        init = tf.global_variables_initializer() #不存在就初始化变量
        sess.run(init)

你可能感兴趣的:(Programming)