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

作者: yiche (你若安好,便是晴天)   2020-07-02 16:54:55
各位板友好
我有多個TensorFlow模型用class各自包裝(如附圖),不過他們是在同一個計算圖之下,
我現在想用tf.train.Saver() 各自存出模型來,突然之間不知道怎麼存
https://i.imgur.com/XwNtdBd.jpg
有想用varlist來指定要存的目標張量
不過用tf.global_variables()看了一下
有108個變數,頭有點暈
有儲存模型相關經驗的板友,請不吝指教!
作者: mychiux413 (小邱)   2020-07-03 16:45:00
我看人家tf都是直接把global的graph用saver存起來,讀回來的時候再用get_tensor指定你要的name如features把layer取出來,所以你class裡面要有初始化跟已存在模型的讀取方式tf.train.load_checkpoint讀回來再存回你的s,r,q_你要用class包的話,用tf.keras.Model繼承是官方推的

Links booklink

Contact Us: admin [ a t ] ucptt.com