From 152b099f9245eb9c8f2597da30ae3e11fb5408d8 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Tue, 4 Aug 2020 06:11:00 +0000 Subject: [PATCH] fix config --- doc/doc_ch/config.md | 8 +++++--- doc/doc_en/config_en.md | 8 +++++--- 2 files changed, 10 insertions(+), 6 deletions(-) diff --git a/doc/doc_ch/config.md b/doc/doc_ch/config.md index bee26370..5e579096 100644 --- a/doc/doc_ch/config.md +++ b/doc/doc_ch/config.md @@ -60,6 +60,8 @@ | 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 一致 | +| function(decay) | 设置decay方式 | - | 目前支持cosine_decay与piecewise_decay | +| step_each_epoch | 每个epoch包含多少次迭代, cosine_decay时有效 | 20 | 计算方式：total_image_num / (batch_size_per_card * card_size) | +| total_epoch | 总共迭代多少个epoch, cosine_decay时有效 | 1000 | 与Global.epoch_num 一致 | +| boundaries | 学习率下降时的迭代次数间隔, piecewise_decay时有效 | - | 参数为列表形式 | +| decay_rate | 学习率衰减系数, piecewise_decay时有效 | - | \ | diff --git a/doc/doc_en/config_en.md b/doc/doc_en/config_en.md index ffead1ee..66578424 100644 --- a/doc/doc_en/config_en.md +++ b/doc/doc_en/config_en.md @@ -60,6 +60,8 @@ 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 | 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 | +| function(decay) | Set the decay function | cosine_decay | Support cosine_decay and piecewise_decay | +| step_each_epoch | The number of steps in an epoch. Used in cosine_decay | 20 | Calculation ：total_image_num / (batch_size_per_card * card_size) | +| total_epoch | The number of epochs. Used in cosine_decay | 1000 | Consistent with Global.epoch_num | +| boundaries | The step intervals to reduce learning rate. Used in piecewise_decay | - | The format is list | +| decay_rate | Learning rate decay rate. Used in piecewise_decay | - | \ | -- GitLab