1. 程式人生 > >tensorflow在訓練過程中逐步減少學習率

tensorflow在訓練過程中逐步減少學習率


# 宣告學習率為不可訓練
learning_rate = tf.Variable(float(learning_rate), trainable=False, dtype=tf.float32)
# 學習率遞減操作,這裡表示每次學習率變成上一次的0.9倍
learning_rate_decay_op = learning_rate.assign(learning_rate * 0.9)

#訓練過程中根據loss情況來執行學習率遞減操作
if ....:  #這裡設定一個需要衰減學習率的條件
    sess.run(learning_rate_decay_op)