超级会员免费看
这篇博客介绍了如何在Detectron2中实现训练过程中计算验证集的损失。通过修改配置，将训练数据替换为验证数据，并利用训练过程中的损失计算方法，可以实时获取验证集的损失。博主提供了具体的代码实现，包括`ValidationLoss`类的定义及其使用方法，展示了如何注册hook并在训练过程中计算`total_val_loss`和各类别的损失。
写在前面的话
该问题在 GitHub的 detectron2 的 issues 上被提出，有人解决了（如下图所示）
提示一下，去 GitHub 上的 issues 搜索问题，尽量找【closed】标签的，这些基本都是有解决方法的问题。
这里只做个记录，仅供学习使用
参考GitHub链接：
How do I compute validation loss during training?
这个实现的很巧妙，直接把训练集的替换成验证集，用原本的训练集的计算loss的方法做计算
添加的包
from detectron2.engine import</
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
