前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: bluemandora
loss = loss1 + loss2 opt = fluid.optimizer.SGD() opt.minimize(loss)
多任务训练可以这样写多个loss求和,但是如果一部分数据只需要反向传播loss1,另一部分数据只需要反向传播loss2,应该怎么写呢