技术栈

lightning

喝过期的拉菲
5 小时前
pytorch·可视化·lightning·指标追踪
使用 Pytorch Lightning 时追踪指标和可视化指标本文介绍了PyTorch Lightning中指标追踪和可视化的方法。主要内容包括:1)使用self.log和self.log_dict记录训练指标;2)通过设置prog_bar=True在命令行进度条显示指标;3)使用TensorBoard可视化指标变化曲线;4)验证集和测试集指标的自动聚合功能,支持mean、min、max等聚合方式;5)通过default_root_dir参数自定义日志保存路径。这些功能为模型训练过程提供了便捷的性能监控和分析手段。
Sylvan Ding
2 个月前
人工智能·pytorch·python·lightning
PyTorch Lightning实战 - 训练 MNIST 数据集利用 PyTorch Lightning 训练 MNIST 数据。验证梯度范数、学习率、优化器对训练的影响。
牛andmore牛
2 年前
fabric·lightning·pytorch多gpu·pytorch混合精度
2、基于pytorch lightning的fabric实现pytorch的多GPU训练和混合精度功能输出结果:以上fabric对应lightning2.1版本,该工具还在开发中,后期会有其他功能。