张量流对象检测API中的超参数优化

时间:2018-08-03 08:55:05

标签: tensorflow object-detection object-detection-api hyperparameters

有什么方法可以在对象检测API的配置文件中指定超参数优化(例如Hyperopt或其他)来微调模型?

2 个答案:

答案 0 :(得分:2)

您可以更改以下用于对象检测的超参数:

  • 学习率
  • 动量
  • 优化器
  • 衰减因子
  • matched_threshold
  • unmatched_threshold

答案 1 :(得分:0)

如何更改配置的优化器

例如,以下是ssd_coco_mobilenetv2的配置文件

“ train_config:{     batch_size:4     优化器{     rms_prop_optimizer:{     learning_rate:{     exponential_decay_learning_rate {
    initial_learning_rate:0.0001     衰减步长:800720     衰减系数:0.95     }     }     动量_最优化值:0.9     衰减:0.9     ε:1.0     }     }     }“

我如何将优化器更改为adagrad而不是rms prop