Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
b27207ba
Commit
b27207ba
authored
Jun 12, 2020
by
tink2123
Browse files
update cosine decay doc
parent
ec35c94b
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
7 additions
and
2 deletions
+7
-2
doc/doc_ch/config.md
doc/doc_ch/config.md
+1
-1
doc/doc_en/config_en.md
doc/doc_en/config_en.md
+1
-1
ppocr/optimizer.py
ppocr/optimizer.py
+5
-0
No files found.
doc/doc_ch/config.md
View file @
b27207ba
...
...
@@ -58,6 +58,6 @@
| beta1 | 设置一阶矩估计的指数衰减率 | 0.9 |
\
|
| beta2 | 设置二阶矩估计的指数衰减率 | 0.999 |
\
|
| decay | 是否使用decay |
\
|
\
|
| function(decay) | 设置decay方式 | cosine_decay |
\
|
| function(decay) | 设置decay方式 | cosine_decay |
目前只支持cosin_decay
|
| step_each_epoch | 每个epoch包含多少次迭代 | 20 | 计算方式:total_image_num / (batch_size_per_card
*
card_size) |
| total_epoch | 总共迭代多少个epoch | 1000 | 与Global.epoch_num 一致 |
doc/doc_en/config_en.md
View file @
b27207ba
...
...
@@ -58,6 +58,6 @@ Take `rec_icdar15_train.yml` as an example:
| beta1 | Set the exponential decay rate for the 1st moment estimates | 0.9 |
\
|
| beta2 | Set the exponential decay rate for the 2nd moment estimates | 0.999 |
\
|
| decay | Whether to use decay |
\
|
\
|
| function(decay) | Set the decay function | cosine_decay |
\
|
| function(decay) | Set the decay function | cosine_decay |
Only support cosine_decay
|
| step_each_epoch | The number of steps in an epoch. | 20 | Calculation :total_image_num / (batch_size_per_card
*
card_size) |
| total_epoch | The number of epochs | 1000 | Consistent with Global.epoch_num |
ppocr/optimizer.py
View file @
b27207ba
...
...
@@ -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
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment