返回信息流同一个model有两个输出,如下所示:
out_1 = model(character_1)
out_2 = model(character_2)
在写loss函数的时候,想写成这种形式:
loss={'out_1': 'categorical_crossentropy', 'out_2':'categorical_crossentropy'},
loss_weights={'out_1': 0.3, 'out_2': 0.3}
该怎么办呢?
out_1 = model(character_1, name='out_1')是报错的
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #35907同步于 2019/12/18
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖
【问题】keras model输出命名问题
Jonathans
2019/12/18镜像同步1 回复
订阅后,新回复会通过你的通知中心匿名送达。
1 条回复
temp_1 = model(character_1)
temp_2 = model(character_2)
out_1 = keras.layers.Lambda(lambda x:x, name='out_1')(temp_1)
out_2 = keras.layers.Lambda(lambda x:x, name='out_2')(temp_2)
解决了[ema3]