[tf]学习率指数衰减

2019-01-16  本文已影响1人  VanJordan
global_step = tf.Variable(0)  
learning_rate = tf.train.exponential_decay(0.1, global_step, 100, 0.96, staircase=True)     #生成学习率  
learning_step = tf.train.GradientDescentOptimizer(learning_rate).minimize(....., global_step=global_step)  #使用指数衰减学习率  
上一篇 下一篇

猜你喜欢

热点阅读