【发布时间】:2016-10-30 03:29:53
【问题描述】:
我正在使用 Tensorflow 0.8 来训练深度神经网络。目前,我遇到一个问题,我想定义两个完全相同的神经网络 N1 和 N2,并且我训练 N1,在训练循环期间,我每 4 次迭代将更新的权重从 N1 复制到 N2。事实上,我知道有办法使用tf.train.saver.save() 将所有N1 权重保存到磁盘上的.ckpt 文件中,并使用tf.train.saver.restore() 从.ckpt 文件中加载这些权重,这相当于复制功能。但是,这种加载/重新加载会影响训练速度,我想知道是否还有其他更有效的方法来进行复制(例如,进行内存复制等)。谢谢!
【问题讨论】:
-
看看我的问题,在这里:stackoverflow.com/questions/48547688/…
标签: python-3.x neural-network tensorflow