时间戳

首页 > 解决方案 > 使用 LearningRateScheduler 实现 cosine_decay

keras - 使用 LearningRateScheduler 实现 cosine_decay

问题描述

我想用来LearningRateScheduler实现余弦衰减学习率，每当我实现以下代码时，它都会给出以下错误：

ValueError：“schedule”函数的输出应该是浮点数。

class CosineDecreaseX(object):
    def __init__(self,  decay_steps , baseLR):
      self.decay_steps = decay_steps
      self.baseLR = baseLR

   def lr_func(self, epoch):
      LR = tf.compat.v1.train.cosine_decay(learning_rate=self.baseLR,global_step= epoch,decay_steps= self.decay_steps)
      return LR

 decay = CosineDecrease( decay_steps=1000 ,  baseLR= 0.0001)
 reduce_lr = tf.compat.v1.keras.callbacks.LearningRateScheduler(decay.lr_func, verbose=1)

谁能告诉我我的代码有什么问题？

标签： keras

解决方案

推荐阅读