update cosine decay doc
@@ -15,6 +15,9 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import paddle.fluid as fluid
+from ppocr.utils.utility import initial_logger
+
+logger = initial_logger()
def AdamDecay(params, parameter_list=None):
@@ -38,6 +41,8 @@ def AdamDecay(params, parameter_list=None):
learning_rate=base_lr,
step_each_epoch=step_each_epoch,
epochs=total_epoch)
+ else:
+ logger.info("Only support Cosine decay currently")
optimizer = fluid.optimizer.Adam(
learning_rate=base_lr,
beta1=beta1,
