From ecbd574e6dadf6a1b6ecdf0857b5039096520df0 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Thu, 16 Dec 2021 17:33:26 +0000 Subject: [PATCH] fix warmup --- configs/rec/ch_ppocr_v2.0/rec_chinese_common_train_v2.0.yml | 1 + configs/rec/ch_ppocr_v2.0/rec_chinese_lite_train_v2.0.yml | 1 + 2 files changed, 2 insertions(+) diff --git 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 index c76063d5..3e427b63 100644 --- 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 diff --git 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 index 563ce110..abd5cd9f 100644 --- 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 -- GitLab