我正在使用 TensorFlow 來訓練神經網絡。這就是我初始化 GradientDescentOptimizer
的方式:
init = tf.initialize_all_variables() sess = tf.Session() sess.run(init) mse = tf. reduce_mean(tf.square(out - out_)) train_step = tf.train.GradientDescentOptimizer(0.3).minimize(mse)
這裡的事情是我不知道怎麼設置學習率的更新規則或衰減值。
如何在這裡使用自適應學習率?