使用 detach()
断开与共享特征层的连接
🚦 梯度流示意:
+---[分类头]------> Loss1 (梯度回传)/
[输入]--[共享特征]--+---[边框头]------> Loss2 (梯度回传)\+---[关键点头]-----> Loss3 (无梯度回传,自身参数冻结)
关键点头 (第三个检测头) 的参数会不会更新?
答案:
是的,第三个检测头(关键点头)的参数依然会更新,即使使用 detach()
断开与共享特征层的连接。
为什么关键点头的参数依然更新?
1. 梯度计算机制: