自动驾驶多任务模型设计思路的简介

1、在自动驾驶领域如何设计一个多任务算法的框架,推动对整体功能上线很重要。

2、一个batch里a任务跑完求解loss_a,再 b任务跑完求解loss_b,接着c任务跑完求解loss_c,。。。。,

3、每个任务的neck和head单独更新梯度和权重,backbone一次batch只更新一次梯度和权重。

4、每个任务单独构建自己的计算图,模型forward backward多次,

5、每次forward backward只处理一个任务,然后做梯度累加,最后统一更新参数。

 

 

 

 

你可能感兴趣的:(深度学习)