且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

如何在 tensorflow 中保存训练好的模型?

更新时间:2023-12-02 08:58:34

最简单的保存和恢复方法:

Simplest way to save and restore:

保存:

saver = tf.train.Saver(max_to_keep=1) 
with tf.Session() as sess:
    # train your model, then:
    savePath = saver.save(sess, 'someDir/my_model.ckpt')

恢复:

with tf.Session() as sess:
    saver = tf.train.import_meta_graph('someDir/my_model.ckpt.meta')
    saver.restore(sess, pathModel + 'someDir/my_model.ckpt')
    # access a variable from the saved Graph, and so on:
    someVar = sess.run('varName:0')

应该这样做