keras模型在用fit训练的时候,如何给loss乘一个数值(张量)然后BP更新网络参数呢?
1条回答 默认 最新
- theworld666 2021-12-02 20:14关注
建议将模型继承keras.Model来写一个自定义模型,在定义训练步骤里写如何计算损失,这样的模型仍然可以使用fit方法来训练
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
keras模型在用fit训练的时候,如何给loss乘一个数值(张量)然后BP更新网络参数呢?
建议将模型继承keras.Model来写一个自定义模型,在定义训练步骤里写如何计算损失,这样的模型仍然可以使用fit方法来训练