fix warmup

2021-12-16 17:33:26 +00:00 · 2021-12-16 17:33:26 +00:00 · ecbd574e6d
parent ff185e6004
commit ecbd574e6d
2 changed files with 2 additions and 0 deletions
--- a/configs/rec/ch_ppocr_v2.0/rec_chinese_common_train_v2.0.yml
+++ b/configs/rec/ch_ppocr_v2.0/rec_chinese_common_train_v2.0.yml
@ -28,6 +28,7 @@ Optimizer:
  lr:
    name: Cosine
    learning_rate: 0.001
+    warmup_epoch: 5
  regularizer:
    name: 'L2'
    factor: 0.00004
--- a/configs/rec/ch_ppocr_v2.0/rec_chinese_lite_train_v2.0.yml
+++ b/configs/rec/ch_ppocr_v2.0/rec_chinese_lite_train_v2.0.yml
@ -28,6 +28,7 @@ Optimizer:
  lr:
    name: Cosine
    learning_rate: 0.001
+    warmup_epoch: 5
  regularizer:
    name: 'L2'
    factor: 0.00001