美文网首页
[tf]学习率指数衰减

[tf]学习率指数衰减

作者: VanJordan | 来源:发表于2019-01-16 20:46 被阅读1次
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)  #使用指数衰减学习率  
  • staircase = True:表示使用指数衰减是阶梯式的,只在特定的步长进行衰减。
    image.png

相关文章

网友评论

      本文标题:[tf]学习率指数衰减

      本文链接:https://www.haomeiwen.com/subject/khsvdqtx.html