问题
想要把PaddleDetection(ppdet)的mask-rcnn固化backbone或者其他部分,然后用新的数据集训练做迁移学习,官方文档目前没有找到解决方案。
处理方法
ppdet官方的回答如下,在需要冻结的网络层的输出部分加入如下代码。
out.stop_gradient = True
目前ppdet并没有一个统一的控制配置,可以自己对框架进行二次开发,添加全局控制变量。
想了解ppdet项目结构请回看这篇文章。
PaddleDetection项目代码结构简介