如何在一个训练中使用多个损失函数
收藏
我用的是v2 0.13.0版本的怎么才能同时监控两个损失函数,
sim_cost = paddle.layer.smooth_l1_cost(prediction, lbl)
cost1 = paddle.layer.classification_cost(name="cost1",input=out1, label=cass_lbl1)
cost2 = paddle.layer.classification_cost(name="cost2",input=out2, label=cass_lbl2)
avg_cost = sim_cost + 0.5 * cost1 + 0.5 * cost2
我使用这样的方式,但是一直训练结果不收敛
0
收藏
请登录后评论
fluid版本倒是可以