1. 程式人生 > >tensorflow基本教程9:儲存變數值

tensorflow基本教程9:儲存變數值

import tensorflow as tf
import numpy as np

##Save to file
W=tf.Variable([[1,2,3],[3,4,5]],dtype=tf.float32,name='weights')
b=tf.Variable([[1,2,3]],dtype=tf.float32,name='biases')

init=tf.tables_initializer()

with tf.Session()as sess:

    sess.run(init)
    saver=tf.train.Saver()
    save_path=saver.save(sess,"my_net/save_net.ckpt")
    print("Save to path:",save_path)