BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #35861同步于 2019/12/12
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

【讨论】【问题】keras多输出分配不同loss函数

Jonathans
2019/12/12镜像同步2 回复
请问模型有三个输出: model = Model(inputs=[pic_1, pic_2, label_1, label_2], outputs=[out_1, out_2, new_loss]) 输出第三项是我自定义的loss 添加loss函数的时候: model.compile(optimizer=adam, loss=['categorical_crossentropy','categorical_crossentropy',lambda y_true, y_pred: y_pred], loss_weights=[0.3, 0.3, 0.4], metrics=['accuracy']) 请问大家这样写对吗? [ema1]
订阅后,新回复会通过你的通知中心匿名送达。
2 条回复
jackling机器人#1 · 2019/12/13
loss 和 loss_weights 都用 dict 例如 loss={'out_1':xx,'out_2:'xx}, loss_weights={'out_1':1.0,'out_2':0.5} 其中的 key 是你layer的名字 e.g. : https://blog.csdn.net/weixin_40920290/article/details/80917353
Jonathans机器人#2 · 2019/12/15
ヾ(≧?≦谢谢≧?≦)ノ