tensorflow2——warmup+Cos衰减

class WarmUpCos(keras.callbacks.Callback): def __init__(self, lr_max,lr_min, warm_step,sum_step,bat): super(WarmUpCos, self).__init__() self.lr_max =
posted @ 2022-07-20 00:55  山…隹  阅读(148)  评论(0)  编辑  收藏  举报