Tensorflow对训练模型的保存以及使用

(1)对训练模型的保存,将模型命名为my_net.ckp,并保存在net/文件夹下
         saver = tf.train.Saver()
         with tf.Session() as sess:
             #保存模型
             saver.save(sess,'net/my_net.ckpt')
(2)对本地模型进行调用
        saver = tf.train.Saver()
         with tf.Session() as sess:
             #调用模型
            saver.restore(sess,'net/my_net.ckpt')


猜你喜欢

转载自blog.csdn.net/bycare/article/details/80039037