[問題] TensorFlow網路參數儲存問題

看板Python作者Paudse (HI)時間6年前 (2018/04/23 19:44)推噓6(6推 0噓 24→)

留言30則, 4人參與討論串1/1

我想將NN的參數儲存下次繼續學習但發現儲存時似乎發生問題叫出的參數每次都一樣我的程式結構如下還請強者指教謝謝 class DQ: def __init__(): self.sess = tf.Session() saver = tf.train.Saver() self.sess.run(tf.global_variables_initializer()) with tf.Session() as sess: if os.path.isfile("save_net.ckpt.index"): saver.restore(sess, "save_net.ckpt") print('File exists, loading previous data!') else: # save_path = self.saver.save(self.sess, "save_net.ckpt") print('File does not exist, starting fresh') def _build_net(self): 省略 def learn(self,save_step): saver = tf.train.Saver() with tf.Session() as sess: sess.run(tf.global_variables_initializer()) save_path = saver.save(sess, "save_net.ckpt") print('save parameters') -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 140.112.53.213 ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1524483897.A.B1B.html ※ 編輯: Paudse (140.112.53.213), 04/23/2018 19:45:34 ※ 編輯: Paudse (140.112.53.213), 04/23/2018 19:46:19

推

jameszhan

04/24 08:25, 6年前 , 1^F

04/24 08:25, 1^F

→

jameszhan

04/24 08:25, 6年前 , 2^F

04/24 08:25, 2^F

推

jameszhan

04/24 08:35, 6年前 , 3^F

04/24 08:35, 3^F