keras模型在用fit训练的时候,如何给loss乘一个数值(张量)然后BP更新网络参数呢?
问题描述:
keras模型在用fit训练的时候,如何给loss乘一个数值(张量)然后BP更新网络参数呢?
答
建议将模型继承keras.Model来写一个自定义模型,在定义训练步骤里写如何计算损失,这样的模型仍然可以使用fit方法来训练