Tensorflow复制权重问题

时间:2016-06-28 00:57:09

标签: python-3.x neural-network tensorflow

我正在使用Tensorflow 0.8训练深度神经网络。目前,我遇到了一个问题,我想要定义两个完全相同的神经网络N1和N2,并且训练N1,在训练循环期间,我每4次迭代将更新的权重从N1复制到N2。事实上,我知道有一种方法可以使用tf.train.saver.save()将所有N1权重保存到磁盘上的.ckpt文件中,并使用tf.train.saver.restore()从.ckpt文件中加载这些权重,这相当于复制功能。但是,这种加载/重新加载会影响训练速度,我想知道是否还有其他更有效的方法来进行复制(例如,进行内存中复制等)。谢谢!

1 个答案:

答案 0 :(得分:0)

如果您可以在此处获得您的代码/更多细节,那将是有益的。但是,您可以返回正在使用的会话来训练N1并在您想要训练N2时访问它。