Tensorflow2.0 自定义余弦退火学习率衰减策略

项目说明

余弦退火学习率衰减的学习率曲线如下图所示:
Tensorflow2.0 自定义余弦退火学习率衰减策略_第1张图片
除了在训练过程中进行学习率衰减以防止求得的最优解不断震荡之外,余弦退火衰减策略规定,学习率每次衰减到0之后,会迅速回升至初始值,这样可以避免陷入局部最优解。关于余弦退火学习率衰减策略的详细介绍可以参考文章:学习率衰减之余弦退火(CosineAnnealing)。

代码实现

在 Tensorflow2.0 中自定义学习率衰减策略其实很简单,可以参考文章:Tensorflow2.0学习率衰减详细汇总。

import tensorflow as tf

你可能感兴趣的:(PHM项目实战--建模篇,深度学习,Tensorflow,2.0,tensorflow,python,深度学习)