讲师中心 微信公众号
AI工具推荐 视频效率加速

如何用Python实现自定义损失函数_继承tf.keras.losses并编写call方法

秋芳君_1680

秋芳君_1680

发布时间:2026-04-06 14:57:34

|

610人浏览过

|

来源于php中文网

原创

必须重写 call 而非 call__,因基类已实现 __call 来处理样本权重与 reduction;call 需专注逐样本损失计算,签名固定为 call(self, y_true, y_pred, sample_weight=None),返回 (batch_size,) 张量。

如何用python实现自定义损失函数_继承tf.keras.losses并编写call方法

继承 tf.keras.losses.Loss 时为什么必须重写 call 而不是 __call__

因为 tf.keras.losses.Loss 基类已实现 __call__,它负责自动处理样本权重、reduction 等通用逻辑;你只需专注计算“原始逐样本损失”,这个逻辑必须放在 call 方法里。否则模型训练会静默跳过你的计算,或报 NotImplementedError。

常见错误现象:
• 模型 loss 值恒为 0 或 nan
• TypeError: call() missing 1 required positional argument: 'y_pred'
• 自定义逻辑完全没生效(比如打印调试语句不触发)

  • 必须用 class MyLoss(tf.keras.losses.Loss) 形式继承,不能只写函数
  • call 方法签名固定为 call(self, y_true, y_pred, sample_weight=None);sample_weight 参数必须显式声明(即使不用),否则 Keras 内部调用会出错
  • 返回值必须是 shape 为 (batch_size,) 的 tensor,不能是标量(如 tf.reduce_mean(...))——那是 reduction 阶段的事

y_true 和 y_pred 的 shape 与 dtype 容易对不齐

尤其在多任务、序列、分割等场景下,Keras 不会帮你 reshape 或 cast,错位会导致 silent broadcast 错误或梯度为 0。

使用场景举例:做带 mask 的序列标注损失,y_true 是 (batch, seq_len) 的 int 类别索引,y_pred 是 (batch, seq_len, num_classes) 的 logits —— 你得自己用 tf.gather_nd 或 tf.one_hot 对齐维度。

立即学习“Python免费学习笔记(深入)”;

python 查询技能
python 查询技能

查询客流数据,输出JSON格式,可直接导入Bitable等可视化工具

下载
  • 务必检查 y_true.dtype:分类任务常用 tf.int32,但某些数据加载器可能输出 float32,导致 tf.nn.sparse_softmax_cross_entropy_with_logits 报错
  • 如果 y_pred 是 logits(未 softmax),别手快写 tf.keras.losses.categorical_crossentropy(y_true, y_pred, from_logits=True) —— 这是函数式 API,和自定义类的 call 无关;你要直接调用底层 op,比如 tf.nn.softmax_cross_entropy_with_logits
  • 涉及 mask 时,不要在 call 外做 mask 过滤(如 y_true[mask]),这会破坏 batch 维度一致性,改用 tf.where + tf.scatter_nd 或乘以 float mask

如何安全地加正则项或辅助 loss 到自定义 call

不能直接在 call 里调用 model.add_loss(),那属于图构建阶段逻辑,而 call 是前向执行阶段。强行调用会报 ValueError: No gradients provided for any variable。

正确做法是把辅助项作为额外输出,在训练 step 中统一处理;或者更简单:用 tf.keras.losses.Loss 的子类机制本身不支持“动态加 loss”,所以辅助项得塞进主 loss 计算里。

  • 想加 L2 正则惩罚?别在 call 里算 tf.nn.l2_loss(model.trainable_variables) —— 变量列表在 call 里不可见;应改用 model.add_loss() 在 call 外、train_step 内调用
  • 若只是简单加一个标量系数项(如对比损失里的 margin),可写成:base_loss + 0.1 * tf.maximum(0.0, margin - similarity),确保所有张量 shape 兼容
  • 注意梯度流:所有参与计算的中间变量必须来自 y_true 或 y_pred,否则梯度断掉。例如用 tf.random.normal 生成噪声参与 loss 计算,会导致反向传播失败

验证自定义 loss 是否真正参与梯度更新

最直接的方式不是看 loss 值下降,而是检查对应层权重是否真的在变 —— 很多时候 loss 输出正常,但梯度根本没传到前面层。

性能影响:自定义 call 中避免 Python for 循环、print、np.xxx 调用,全部换成 tf ops;否则会强制退出 graph mode,退化为 eager 执行,训练慢几倍。

  • 加一行 tf.debugging.check_numerics(y_pred, 'y_pred has nan') 在 call 开头,能早发现 inf/nan 污染
  • 用 tf.GradientTape 单独测:with tf.GradientTape() as t: loss = my_loss(y_true, y_pred); grads = t.gradient(loss, model.trainable_variables),确认 grads 列表里没有 None
  • 如果 loss 依赖外部状态(如 moving average),别存成 self.xxx 实例变量 —— Keras 可能跨 replica 复制对象,状态不同步;改用 tf.Variable 并设 aggregation=tf.VariableAggregation.MEAN

事情说清了就结束

热门AI工具

更多
豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1711

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4304

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1689

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

25117

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3067

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3087

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1163

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

PixTV官网入口地址合集
PixTV官网入口地址合集

本专题汇总了 PixTV AI 一站式视频创作平台的官方入口与使用教程。无需下载软件,浏览器直接访问即可使用。平台将剧本、图像、视频、声音与剪辑整合在“无限画布”中,接入 GPT Image 2.5、Seedance 2.5 等头部模型。本专题整理了从新建画布、角色锚定、分镜拆分到视频生成与导出的完整操作指南,助你快速上手 AI 短剧与漫剧创作。

20

2026.10.10

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn