[讨论] 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