PTT數位生活區 / Python

[討論] TensorFlow 1.x 計算圖多模型儲存

看板Python作者 (陽光灑肩頭 彷若自由人)時間5年前發表 (2020/07/02 08:54), 5年前編輯推噓1(1推 0噓 5→)
留言6則, 1人參與, 5年前最新討論串1/1
各位板友好 我有多個TensorFlow模型用class各自包裝(如附圖)，不過他們是在同一個計算圖之下， 我現在想用tf.train.Saver() 各自存出模型來，突然之間不知道怎麼存 https://i.imgur.com/XwNtdBd.jpg
有想用varlist來指定要存的目標張量 不過用tf.global_variables()看了一下 有108個變數，頭有點暈 有儲存模型相關經驗的板友，請不吝指教！ -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 220.138.50.110 (臺灣) ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1593680097.A.7D6.html ※ 編輯: yiche (220.138.50.110 臺灣), 07/02/2020 16:58:16
→
07/03 16:45, 5年前 , 1F
我看人家tf都是直接把global的graph用saver存起來,讀
07/03 16:45, 1F
→
07/03 16:46, 5年前 , 2F
回來的時候再用get_tensor指定你要的name如features
07/03 16:46, 2F
→
07/03 16:47, 5年前 , 3F
把layer取出來,所以你class裡面要有初始化跟已存在模
07/03 16:47, 3F
→
07/03 16:48, 5年前 , 4F
型的讀取方式
07/03 16:48, 4F
→
07/03 16:51, 5年前 , 5F
tf.train.load_checkpoint讀回來再存回你的s,r,q_
07/03 16:51, 5F
推
07/03 16:56, 5年前 , 6F
你要用class包的話，用tf.keras.Model繼承是官方推的
07/03 16:56, 6F
謝謝m大 我再摸索看看 ※ 編輯: yiche (220.138.50.3 臺灣), 07/04/2020 08:57:06
‣ 更多 yiche 的文章
文章代碼(AID): #1U_Q3XVM (Python)
Python 近期熱門文章
更多 近期熱門文章 >>
PTT數位生活區 即時熱門文章
更多 即時熱門文章 >>
‣ 更多 yiche 的文章
文章代碼(AID): #1U_Q3XVM (Python)