重置模型中的所有参数,但TensorFlow中的变量除外

时间:2018-08-15 16:40:28

标签: tensorflow

我有很深的CNN,经过一段时间的训练后,我想将网络的所有参数重置为原始值,除了一些嵌入。

我尝试执行以下操作

tempEmbeds = deepNet.W
deepNet = DeepNet(params=...) # create a new graph using a constructor
sess.run(tf.global_variables_initializer())    
sess.run(deepNet.embeds.assign(tempEmbeds))

但是我内存不足,还有其他方法或想法可以轻松地做到这一点吗?

0 个答案:

没有答案