作者yiche (阳光洒肩头 彷若自由人)
看板Python
标题[讨论] TensorFlow 1.x 计算图多模型储存
时间Thu Jul 2 16:54:55 2020
各位板友好
我有多个TensorFlow模型用class各自包装(如附图),不过他们是在同一个计算图之下,
我现在想用tf.train.Saver() 各自存出模型来,突然之间不知道怎麽存
https://i.imgur.com/XwNtdBd.jpg
有想用varlist来指定要存的目标张量
不过用tf.global_variables()看了一下
有108个变数,头有点晕
有储存模型相关经验的板友,请不吝指教!
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 220.138.50.110 (台湾)
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1593680097.A.7D6.html
※ 编辑: yiche (220.138.50.110 台湾), 07/02/2020 16:58:16
1F:→ mychiux413: 我看人家tf都是直接把global的graph用saver存起来,读07/03 16:45
2F:→ mychiux413: 回来的时候再用get_tensor指定你要的name如features07/03 16:46
3F:→ mychiux413: 把layer取出来,所以你class里面要有初始化跟已存在模07/03 16:47
4F:→ mychiux413: 型的读取方式07/03 16:48
5F:→ mychiux413: tf.train.load_checkpoint读回来再存回你的s,r,q_07/03 16:51
6F:推 mychiux413: 你要用class包的话,用tf.keras.Model继承是官方推的07/03 16:56
谢谢m大 我再摸索看看
※ 编辑: yiche (220.138.50.3 台湾), 07/04/2020 08:57:06