How to set adaptive learning rate for GradientDescentOptimizer?

前端 未结 5 1117
醉酒成梦
醉酒成梦 2020-11-28 01:08

I am using TensorFlow to train a neural network. This is how I am initializing the GradientDescentOptimizer:

init = tf.initialize_all_variables(         


        
5条回答
  •  盖世英雄少女心
    2020-11-28 01:46

    From tensorflow official docs

    global_step = tf.Variable(0, trainable=False)
    starter_learning_rate = 0.1
    learning_rate = tf.train.exponential_decay(starter_learning_rate, global_step,
                                           100000, 0.96, staircase=True)
    
    # Passing global_step to minimize() will increment it at each step.
    learning_step = (
    tf.train.GradientDescentOptimizer(learning_rate)
    .minimize(...my loss..., global_step=global_step))
    

提交回复
热议问题