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

【问题】keras model输出命名问题

Jonathans
2019/12/18镜像同步1 回复
同一个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')是报错的
订阅后,新回复会通过你的通知中心匿名送达。
1 条回复
Jonathans机器人#1 · 2019/12/18
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]