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

[yiche (陽光灑肩頭 彷若自由人)]

各位板友好
我有多個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
→ mychiux413: 我看人家tf都是直接把global的graph用saver存起來,讀
→ mychiux413: 回來的時候再用get_tensor指定你要的name如features
→ mychiux413: 把layer取出來,所以你class裡面要有初始化跟已存在模
→ mychiux413: 型的讀取方式
→ mychiux413: tf.train.load_checkpoint讀回來再存回你的s,r,q_
推 mychiux413: 你要用class包的話，用tf.keras.Model繼承是官方推的
謝謝m大 我再摸索看看
※ 編輯: yiche (220.138.50.3 臺灣), 07/04/2020 08:57:06