首页 > 解决方案 > 我应该在张量板上观察哪些损失标量?

问题描述

我正在训练只检测人的自定义对象检测模型。
我遵循了这个 github 文档
这是我的环境。

Tensorflow version : 1.13.1 GPU
Tensorboard version : 1.13.1
Pre-trained model : SSD mobilenet v2 quantized 300x300 coco

根据上面的文件,我的模型的损失应该下降到 2 以下。
所以我打开了张量板,发现分类损失、总损失、克隆损失标量。
这是我的张量板快照。
我应该查找哪些损失标量?
我也有收敛的问题。
我的模型的损失收敛在 4,而不是 2。
我该如何解决?
我将学习率 0.004(提供值)降低到 0.0000095。
但它仍然在 4 处收敛。
这是我的 train_config。

train_config: {
  batch_size: 6
  optimizer {
    rms_prop_optimizer: {
      learning_rate: {
        exponential_decay_learning_rate {
          initial_learning_rate: 0.0000095
          decay_steps: 500
          decay_factor: 0.95
        }
      }
      momentum_optimizer_value: 0.9
      decay: 0.9
      epsilon: 1.0
    }

有什么要改变的吗?
感谢所有帮助。

标签: tensorflowobject-detectiontensorboardobject-detection-api

解决方案


推荐阅读