美文网首页
keras定期调整学习率

keras定期调整学习率

作者: 你说你要一场 | 来源:发表于2019-07-18 18:20 被阅读0次

keras可以import两个函数
定义一个函数
进而实现动态调整学习率。
并且fit里的callbacks中的学习率会覆盖掉complie里的学习率。
还有,def scheduler(epoch):里的model要是里接下来要引用的model,当有多个model或者model被重新命名是需要注意这一点。
下面是代码:

import keras.backend as K
from keras.callbacks import LearningRateScheduler
def scheduler(epoch):
    if epoch<1000:
        # lr=K.get_value(model2.optimizer.lr)
        K.set_value(model2.optimizer.lr, 0.00001)
        return K.get_value(model2.optimizer.lr)
    else:
        K.set_value(model2.optimizer.lr, 0.0000001)
        return K.get_value(model2.optimizer.lr)
reduce_lr = LearningRateScheduler(scheduler)
model2 = Sequential()
    model2.add(Dense(units=1,kernel_initializer="ones",bias_initializer=initializers.Constant(value=-0.01),kernel_constraint=unit_norm(),input_dim=1))
    model2.compile(optimizer=Adam(lr=0.0000001),loss='mse')
    history = model2.fit(test_in,test_out,epochs=2000,verbose=1,callbacks=[reduce_lr])

下图是加不加callbacks的对比:


不加callbacks
加callbacks 不加callbacks
加callbacks

相关文章

  • keras定期调整学习率

    keras可以import两个函数定义一个函数进而实现动态调整学习率。并且fit里的callbacks中的学习率会...

  • Keras使用

    参考:Keras学习率调整深度学习框架Keras使用心得 一、如何调整学习率 Keras提供两种学习率适应方法,可...

  • tensorflow 学习率控制

    为什么调整学习率 学习率的调整 离散下降(discrete staircase)对于深度学习来说,每 tt 轮学习...

  • Keras学习资源汇总

    目录: Keras简介 Keras学习手册 Keras学习视频 Keras代码案例Keras&NLPKeras&C...

  • 如何更清晰全面的了解自己?

    无论工作还是学习,都要定期调整自己的个人状态。 就像一部机器一样的,也需要定期的检查调整。 随时把自己调整到最好的...

  • Pytorch中的学习率衰减方法

    Pytorch 中的学习率调整方法 Pytorch中的学习率调整有两种方式: 直接修改optimizer中的lr参...

  • 动态调整学习率

    学习率对模型的学习效果影响很大。 方法一,使用优化器的decay参数 方法二,使用learningRateSche...

  • 学习率的调整

    1. cosine学习率下降

  • 2018-04-07 深度学习训练中 超参数调整 笔记

    超参数 1 学习率 初始值0.1, 如何调整学习率,准确率不再下降,就减小学习率。准确率下降太慢,就增大学习率 2...

  • Keras学习记录

    Keras学习笔记 keras.io keras.io-zh keras-cn Keras是一个高层神经网络API...

网友评论

      本文标题:keras定期调整学习率

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