当前位置: 首页 > news >正文

lightning.pytorch.callbacks内置的Callbacks介绍

PyTorch Lightning 提供了一些 内置回调 (Callback),可以在训练过程中自动执行 检查点保存、学习率调度、早停 等功能。通过使用 Trainer(callbacks=[...]) 来传入这些回调。

PyTorch Lightning 的 Callback 是一种强大的工具,允许用户在训练过程中插入自定义逻辑,而无需修改核心的训练代码。Callback 的设计基于钩子(hooks),在训练流程的特定点执行自定义代码。

1. 内置回调列表

PyTorch Lightning 内置了以下 Callbacks(可以直接使用,无需自定义):

回调名称 功能
ModelCheckpoint 自动保存最佳模型(基于验证指标)
EarlyStopping 自动停止训练(当验证指标不再改善时)
LearningRateMonitor 记录学习率变化(支持 TensorBoard & WandB
RichProgressBar 使用 rich 库美化训练进度条
TQDMProgressBar 默认的 tqdm 进度条
DeviceStatsMonitor 监控 GPU/CPU 使用情况
BatchSizeFinder 自动寻找最优 batch size
GradientAccumulationScheduler 自动调整梯度累积步数
ModelSummary 打印模型结构和参数量
StochasticWeightAveraging (SWA) 使用 SWA 进行权重平均,提高泛化能力

2. 详细介绍 & 代码示例

(1)ModelCheckpoint - 自动保存最佳模型

用于 自动保存模型检查点(ckpt),可以: ✅ 保存最佳模型(基于某个指标,如 val_loss)。
✅ 定期保存(如每 n 个 epoch 保存一次)。
✅ 限制最大检查点数量(避免磁盘占用过大)。

📜 示例:

from lightning.pytorch.callbacks import ModelCheckpoint

# 仅保存最优模型 (基于 val_loss)
check
http://www.dtcms.com/a/17276.html

相关文章:

  • Nginx 上安装 SSL 证书并启用 HTTPS 访问
  • cpu温度多少正常?cpu温度过高怎么办
  • 7. Docker 容器数据卷的使用(超详细的讲解说明)
  • 编译安装vsftpd后,虚拟用户模式下登录提示:“530 Login incorrect.“
  • 如果维护了多种语言版本的文本对应关系,另一种语言只需要将对应文本填入对应位置即可,这种情况应该如何实现自动填入
  • 【verilog】函数clogb2的解读
  • 【python】http.server内置库构建临时文件服务
  • C语言蓝桥杯1003: [编程入门]密码破译
  • 【第4章:循环神经网络(RNN)与长短时记忆网络(LSTM)——4.2 LSTM的引入与解决长期依赖问题的方法】
  • 第一章嵌入式系统概论考点05数字(电子)文本
  • 计算机网络-八股-学习摘要
  • 【学术投稿】第五届计算机网络安全与软件工程(CNSSE 2025)
  • 一天急速通关SpringMVC
  • 算法08-递归调用转为循环的通用方法
  • Lua闭包的使用以及需要注意的问题
  • RadASM环境,win32汇编入门教程之二
  • 技术评测:MaxCompute MaxFrame——阿里云自研分布式计算框架的Python编程接口
  • 第四十四篇--Tesla P40+Janus-Pro-7B部署与测试
  • CI/CD部署打包方法
  • 2.11寒假
  • SiliconCloud 支持deepseek,送2000w token
  • 使用 Nginx 搭建代理服务器(正向代理 HTTPS 网站)指南
  • 剑指offer第2版:搜索算法(二分/DFS/BFS)
  • 算法练习——哈希表
  • Python实现从SMS-Activate平台,自动获取手机号和验证码(进阶版2.0)
  • 前端包管理器的发展以及Npm、Yarn和Pnpm对比
  • AWTK fscript 中的 TCP/UDP 客户端扩展函数
  • C++课程设计 运动会分数统计(含源码)
  • 打开游戏缺少C++组件怎么修复?缺少C++组件问题的解决方法
  • FastAPI 高并发与性能优化