Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
秋饼
PaddleOCR
提交
b47b5521
P
PaddleOCR
项目概览
秋饼
/
PaddleOCR
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleOCR
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
b47b5521
编写于
6月 15, 2020
作者:
D
Double_V
提交者:
GitHub
6月 15, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #184 from tink2123/add_cosine_decay
add cosine decay
上级
1963d80d
b27207ba
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
49 addition
and
0 deletion
+49
-0
configs/rec/rec_icdar15_train.yml
configs/rec/rec_icdar15_train.yml
+4
-0
doc/doc_ch/config.md
doc/doc_ch/config.md
+15
-0
doc/doc_en/config_en.md
doc/doc_en/config_en.md
+15
-0
ppocr/optimizer.py
ppocr/optimizer.py
+15
-0
未找到文件。
configs/rec/rec_icdar15_train.yml
浏览文件 @
b47b5521
...
@@ -41,3 +41,7 @@ Optimizer:
...
@@ -41,3 +41,7 @@ Optimizer:
base_lr
:
0.0005
base_lr
:
0.0005
beta1
:
0.9
beta1
:
0.9
beta2
:
0.999
beta2
:
0.999
decay
:
function
:
cosine_decay
step_each_epoch
:
20
total_epoch
:
1000
doc/doc_ch/config.md
浏览文件 @
b47b5521
...
@@ -46,3 +46,18 @@
...
@@ -46,3 +46,18 @@
| img_set_dir | 数据集路径 | ./train_data |
\
|
| img_set_dir | 数据集路径 | ./train_data |
\
|
| label_file_path | 数据标签路径 | ./train_data/rec_gt_train.txt|
\
|
| label_file_path | 数据标签路径 | ./train_data/rec_gt_train.txt|
\
|
| infer_img | 预测图像文件夹路径 | ./infer_img |
\|
| infer_img | 预测图像文件夹路径 | ./infer_img |
\|
## 配置文件 Optimizer 系列参数介绍
以
`rec_icdar15_train.yml`
为例
| 字段 | 用途 | 默认值 | 备注 |
| :---------------------: | :---------------------: | :--------------: | :--------------------: |
| function | 选择优化器 | pocr.optimizer,AdamDecay | 目前只支持Adam方式 |
| base_lr | 设置初始学习率 | 0.0005 |
\
|
| beta1 | 设置一阶矩估计的指数衰减率 | 0.9 |
\
|
| beta2 | 设置二阶矩估计的指数衰减率 | 0.999 |
\
|
| decay | 是否使用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
浏览文件 @
b47b5521
...
@@ -46,3 +46,18 @@ Take `rec_chinese_reader.yml` as an example:
...
@@ -46,3 +46,18 @@ Take `rec_chinese_reader.yml` as an example:
| img_set_dir | Image folder path | ./train_data |
\
|
| img_set_dir | Image folder path | ./train_data |
\
|
| label_file_path | Groundtruth file path | ./train_data/rec_gt_train.txt|
\
|
| label_file_path | Groundtruth file path | ./train_data/rec_gt_train.txt|
\
|
| infer_img | Result folder path | ./infer_img |
\|
| infer_img | Result folder path | ./infer_img |
\|
## Introduction to Optimizer parameters of Configuration file
Take
`rec_icdar15_train.yml`
as an example:
| Parameter | Use | Default | None |
| :---------------------: | :---------------------: | :--------------: | :--------------------: |
| function | Select Optimizer function | pocr.optimizer,AdamDecay | Only support Adam |
| base_lr | Set the base lr | 0.0005 |
\
|
| 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 | 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
浏览文件 @
b47b5521
...
@@ -15,6 +15,9 @@ from __future__ import absolute_import
...
@@ -15,6 +15,9 @@ from __future__ import absolute_import
from
__future__
import
division
from
__future__
import
division
from
__future__
import
print_function
from
__future__
import
print_function
import
paddle.fluid
as
fluid
import
paddle.fluid
as
fluid
from
ppocr.utils.utility
import
initial_logger
logger
=
initial_logger
()
def
AdamDecay
(
params
,
parameter_list
=
None
):
def
AdamDecay
(
params
,
parameter_list
=
None
):
...
@@ -28,6 +31,18 @@ def AdamDecay(params, parameter_list=None):
...
@@ -28,6 +31,18 @@ def AdamDecay(params, parameter_list=None):
base_lr
=
params
[
'base_lr'
]
base_lr
=
params
[
'base_lr'
]
beta1
=
params
[
'beta1'
]
beta1
=
params
[
'beta1'
]
beta2
=
params
[
'beta2'
]
beta2
=
params
[
'beta2'
]
if
'decay'
in
params
:
params
=
params
[
'decay'
]
decay_mode
=
params
[
'function'
]
step_each_epoch
=
params
[
'step_each_epoch'
]
total_epoch
=
params
[
'total_epoch'
]
if
decay_mode
==
"cosine_decay"
:
base_lr
=
fluid
.
layers
.
cosine_decay
(
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
(
optimizer
=
fluid
.
optimizer
.
Adam
(
learning_rate
=
base_lr
,
learning_rate
=
base_lr
,
beta1
=
beta1
,
beta1
=
beta1
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录