学习率的改变

  1. import tensorflow as tf;  
  2. import numpy as np;  
  3. import matplotlib.pyplot as plt;  
  4.   
  5. learning_rate = 0.1  
  6. decay_rate = 0.96  
  7. global_steps = 1000  
  8. decay_steps = 100  
  9.   
  10. global_ = tf.Variable(tf.constant(0))  
  11. c = tf.train.exponential_decay(learning_rate, global_, decay_steps, decay_rate, staircase=True)  
  12. d = tf.train.exponential_decay(learning_rate, global_, decay_steps, decay_rate, staircase=False)  
  13.   
  14. T_C = []  
  15. F_D = []  
  16.   
  17. with tf.Session() as sess:  
  18.     for i in range(global_steps):  
  19.         T_c = sess.run(c,feed_dict={global_: i})  
  20.         T_C.append(T_c)  
  21.         F_d = sess.run(d,feed_dict={global_: i})  
  22.         F_D.append(F_d)  
  23.   
  24.   
  25. plt.figure(1)  
  26. plt.plot(range(global_steps), F_D, 'r-')  
  27. plt.plot(range(global_steps), T_C, 'b-')  
  28.       

  1. plt.show()  
  2. 分析:

    初始的学习速率是0.1,总的迭代次数是1000次,如果staircase=True,那就表明每decay_steps次计算学习速率变化,更新原始学习速率,如果是False,那就是每一步都更新学习速率。红色表示False,绿色表示True。


    结果:


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值