运行Adam Optimizer

时间:2017-11-01 19:28:24

标签: python tensorflow conv-neural-network

我试图运行一个AdamOptimizer进行一步训练,不成功。

viewer.impl.

控制台吐出一个难看的错误:

optimizer = tf.train.AdamOptimizer(learning_rate)
init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init)
    sess.run(optimizer.minimize(cost), feed_dict={X:X_data, Y: Y_data})

在代码中,cost是一个明确定义的函数,使用两个参数X,Y(分别为NN和训练标签的条目)实现conv NN加上后勤损失函数

关于什么可能出错的任何想法?

1 个答案:

答案 0 :(得分:16)

optimizer.minimize(cost)正在创造新的价值观图表中的变量。

当您致电sess.run(init)时,.minimize方法创建的变量尚未定义:从此您的错误。

您只需在调用tf.global_variables_initializer()之前声明最小化操作:

optimizer = tf.train.AdamOptimizer(learning_rate)
minimize = optimizer.minimize(cost)
init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init)
    sess.run(minimize, feed_dict={X:X_data, Y: Y_data})