我在哪里可以找到模型使用的损失指标类型？

Posted 2023-03-27

【中文标题】我在哪里可以找到模型使用的损失指标类型？

【英文标题】：Where can I find what type of loss metric a model uses?

【发布时间】：2019-07-11 10:11:42

【问题描述】：

我试图弄清楚 ssdlite_mobilenet_v2_coco 中使用了哪些损失函数，但我似乎无法找到它在 repo 或在线指定的位置。这是我在训练时损失的 Tensorboard，但它没有指定损失指标是什么。

回购链接：https://github.com/tensorflow/models/tree/master/research/object_detection

我正在学习本教程： https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10#8-use-your-newly-trained-object-detection-classifier

我已尝试查看配置文件和代码的某些部分，但没有看到它使用什么类型的指标。

我是否可以在某个位置找到用于存储库中每种模型架构的损失指标？谢谢！

【参考方案1】：

您可以看到config file 中使用的损失函数。损失由两个损失组成：weighted_sigmoid 用于分类，weighted_smooth_l1 用于定位，每个损失的权重为 1。要查看这些损失的代码，您可以查看 loss functions here。请注意，总损失还有另一个用于正则化的组件。

【讨论】：

感谢您的帮助！

当然可以，很高兴为您提供帮助。如果它回答了你的问题，请用 V 设置答案，让其他有相同问题的人知道这个答案是正确的/有帮助的。