1.辅助头训练是一种常用的方法,训练时辅助训练,推理时可以去掉。
辅助头的设计loss与主体头的优化目标不一致。例如:
主体头是giou和分类loss,辅助头可以是软标签,让他的结果与一个大模型结构接近。这样反向传播时影响了主干的参数,最后推理的时候只需要主体头,不会相互影响。
辅助头一般有几种:atss等。。。。
辅助头训练
最新推荐文章于 2024-09-12 18:02:47 发布
1.辅助头训练是一种常用的方法,训练时辅助训练,推理时可以去掉。
辅助头的设计loss与主体头的优化目标不一致。例如:
主体头是giou和分类loss,辅助头可以是软标签,让他的结果与一个大模型结构接近。这样反向传播时影响了主干的参数,最后推理的时候只需要主体头,不会相互影响。
辅助头一般有几种:atss等。。。。