重置Tensorflow Adam优化器

时间:2019-05-22 11:37:36

标签: python tensorflow machine-learning neural-network

我正在使用Tensorflow对神经网络模型执行K折交叉验证,因此我需要对模型进行k次训练。

我想知道如何重设tf.train.AdamOptimizer的状态,以便每次折叠后都可以进行新的训练。

有人建议:

 sess.run(tf.variables_initializer(optimizer.variables()))

但这会引发以下错误:

'Operation' object has no attribute 'variables'

0 个答案:

没有答案