学习率的指数衰减函数 tf.exponetial_decay(a,b,c,d,staircase = Fasle)

通过这个函数,可以已较大的学习率快速得到一个比较优的解,随后随着迭代的继续逐步减少学习率,使得模型在训练后期更加稳定。

计算公式:decay_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps)

decay_learnig : 每轮使用的学习率
learning_rate: 事先设定的初始学习率
decay_rate : 衰减系数
staircase = Fasle: default 默认,以平滑的曲线下降,如果设置为True则,global_step/decay_steps 取整数,阶梯形下降

global_step = tf.Variable(0)

% learning_rate = tf.train.exponential_decay(learning_rate, global_step, one_cycle_step, learning_rate_decay, staircase = Fasle)
% global_step : current iterater 当前的迭代轮数
learning_rate = tf.train.exponential_decay(0.3, global_step, 100, 0.99, staircase = True)
%使用minimize 函数传入global——step 自动更新global——step参数
loss = 需要定义一个loss方程
learning_step = tf.train.GradientDescentOptimizer(learning_rate).minimize(loss, global_step = global_step)
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值