tensorflow - 如何使用 tf.contrib.keras.optimizers.Adamax ？

Question

我在 tensorflow 1.2 中使用 tf.AdamOptimizer:

train_opt = tf.train.AdamOptimizer(learning_rate=0.0004).minimize(error)

现在我想用新的 tf.contrib.keras.optimizers 替换它:

train_opt = tf.contrib.keras.optimizers.Adamax(lr=0.0004)

但是我怎样才能最小化错误呢？？ (添加 .minimize(error) 不起作用)

Answer 1

使用:

train_opt = 
    tf.contrib.keras.optimizers.Adamax().get_updates(loss=cost,constraints=[], params=train_params)

cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels = y_, logits = y))

train_params = network.all_params