tensorflow學習之（十一）將python代碼寫入檔案

標籤：and   variables   ESS   span   def   儲存   run   file   flow   

#save to fileimport tensorflow as tfimport  numpy as np##（1）Save to file 把相關變數儲存到檔案中#remember to define the same dtype and shape when restoreW = 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.initialize_all_variables()saver = tf.train.Saver()with tf.Session() as sess:    sess.run(init)    save_path = saver.save(sess,"my_net/save_net.ckpt")    print("Save to path : ",save_path)##（2）restore variables 從檔案中取出相關變數#redefine the same shape and same type for you variablesW = tf.Variable(np.arange(6).reshape((2,3)),dtype=tf.float32,name="weights")#reshape((2,3):2行3列b = tf.Variable(np.arange(3).reshape((1,3)),dtype=tf.float32,name="biases")#not need init stepsaver = tf.train.Saver()with tf.Session() as sess:    saver.restore(sess,"my_net/save_net.ckpt")    print("weights: ",sess.run(W))    print("biases: ",sess.run(b))

 

tensorflow學習之（十一）將python代碼寫入檔案