首页 Paddle框架 帖子详情
如何在一个训练中使用多个损失函数
收藏
快速回复
Paddle框架 问答深度学习模型训练 3746 1
如何在一个训练中使用多个损失函数
收藏
快速回复
Paddle框架 问答深度学习模型训练 3746 1

我用的是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
收藏
回复
全部评论(1)
时间顺序
夜雨飘零1
#2 回复于2018-12

fluid版本倒是可以

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户