python - 优化器返回 None

Question

我正在尝试运行 AdamOptimizer 进行一步训练，但未成功。

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

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

在代码中，cost 是一个定义明确的函数，它使用两个参数 X、Y(分别是 NN 的入口和训练标签)实现一个 conv NN 和一个 logistic 损失函数

当我运行它时，控制台通知我运行没有返回 None 作为输出，这让我感到困惑，因为我期望它返回我的成本。

我做错了什么？

Answer 1

我不认为优化器 会返回任何东西。 optimizer.minimize(或 train_op)将返回 op它将用于更新可训练权重并增加全局步长。如果你想退回损失(或成本)，那么你必须 specify所以在 sess.run([..., loss, ...], ...)

您的代码可能如下所示:

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

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