TensorFlow：恢复模型

问题描述：

我试图在三维结束时保存模型，并在每次训练开始时恢复模型。我只是跟着this link做了什么。TensorFlow：恢复模型

saver = tf.train.Saver() 
    with tf.Session(graph=graph) as session: 
     # Initializate the weights and biases 
     tf.global_variables_initializer().run() 
     new_saver = tf.train.import_meta_graph('model.meta') 
     new_saver.restore(sess,tf.train.latest_checkpoint('./')) 
     W1 = session.run(W) 
     print(W1) 

     for curr_epoch in range(num_epochs): 
      train_cost = train_ler = 0 
      start = time.time() 
      for batch in range(num_batches_per_epoch): 
       ...Some training... 

     W2 = session.run(W) 
     print(W2) 
     save_path = saver.save(session, "models/model")

但它给下面的错误：

---> new_saver.restore(session, tf.train.latest_checkpoint('./')) 
SystemError: <built-in function TF_Run> returned a result with an error set

谁能帮助我吗？非常感谢！

要保存到“模式/模式”尚未从加载“./” ，你有没有尝试修复路径？ – lejlot

是的，起初我运行代码而不恢复，然后我运行上面的代码。 –

答

如果您要加载./，您必须确保您的控制台（用于启动python程序）实际上设置在该目录（models /）上。但在这种情况下，它会将新数据保存在新目录中。因此，与./models/加载，而不是

（你也并不需要引发变量，还原这是否适合你。）

现在真的有用！谢谢sooooo @Xethoras。祝福你！ –

TensorFlow：恢复模型

相关推荐