PaddleDection2.0版本使用余弦退火
双子zh00 发布于2021-09 浏览:2555 回复:0
0
收藏

在PaddleDection中怎么使用CosineDecay

下面是优化器配置文件中的

LearningRate:
  base_lr: 0.001
  schedulers:
  - !CosineDecay
    max_epochs:1000
    # gamma: 0.1
    # milestones:
    # - 166
    # - 216
  - !LinearWarmup
    start_factor: 0.
    steps: 4000

按照上面直接修改 是错的,

LearningRate:
  base_lr: 0.001
  schedulers:
  - !PiecewiseDecay
      # max_epochs:1000
      
    gamma: 0.1
    milestones:
    - 166
    - 216
  - !LinearWarmup
    start_factor: 0.
    steps: 4000

PiecewiseDecay这样的是对的,不知道是什么原因

收藏
点赞
0
个赞
TOP
切换版块