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

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.kler.cn/a/547070.html

相关文章:

  • LeapMotion第2代 Unity示范代码(桌面开发)
  • DeepSeekApi对接流式输出异步聊天功能:基于Spring Boot和OkHttp的SSE应用实现
  • 1.5 企业级AI大模型四阶技术全景解析:从Prompt到Pre-training的进化路径
  • 23. AI-大语言模型
  • 【AI系列】从零开始学习大模型GPT (2)- Build a Large Language Model (From Scratch)
  • Springboot整合ES
  • Linux系统使用ollama本地安装部署DeepSeekR1 + open-webui
  • 字玩FontPlayer开发笔记14 Vue3实现多边形工具
  • wps接入DeepSeek教程
  • uniapp语音时的动态音波的实现
  • SpringBoot中Mybatis记录执行sql日志
  • 基于 STM32 平台的音频特征提取与歌曲风格智能识别系统
  • TypeScript装饰器 ------- 学习笔记分享
  • DeepSeek 模型部署与使用技术评测(基于阿里云零门槛解决方案)
  • 部署onlyoffice后,php版的callback及小魔改(logo和关于)
  • Java项目引入DeepSeek搭建私有AI
  • React历代主要更新
  • 使用EVE-NG-锐捷实现NAT
  • 尚硅谷爬虫note003
  • 微软AutoGen介绍——Managing State保存并加载持续会话的Agents和Teams