更改张量流对象检测中的优化器

Question

如何更改配置的优化器

例如以下是 ssd_coco_mobilenetv2 的 confgi

train_config: { 
  batch_size: 4 
  optimizer {
    rms_prop_optimizer: { 
      learning_rate: { 
        exponential_decay_learning_rate {
          initial_learning_rate: 0.0001 decay_steps: 800720 decay_factor: 0.95 
        } 
      } 
      momentum_optimizer_value: 0.9 
      decay: 0.9 
      epsilon: 1.0 
    } 
  } 
}

Answer 1

这是与优化器对应的proto文件。根据proto文件，您可以选择三种不同的优化器，例如

1. 均方根属性优化器

2. 动量优化器

3. 亚当优化器

然后对于每个优化器，您可以将学习率配置为以下之一

1. 常数学习率
2. 指数衰减学习率
3. 手动步数学习率
4. 余弦衰减学习率

然后对于每个学习率，您可以配置值是什么，默认值也是由proto文件提供的。