Day48TensorBoard使用

旨在演示如何使用 PyTorch 对预训练的 ResNet-18 模型在 CIFAR-10 数据集上进行微调(Fine-tuning),并利用 TensorBoard 工具对训练过程进行实时监控和可视化。

一、TensorBoard 是什么?

TensorBoard 是 TensorFlow 官方提供的可视化工具,就像给你的深度学习训练过程装了一个 "监控仪表盘"。它能帮你直观地看到:

  • 损失值(loss)、准确率(accuracy)等指标的变化曲线
  • 模型的网络结构
  • 训练过程中参数的分布
  • 图片、文本等数据的可视化
  • 甚至是训练过程中的计算图
  1. TensorBoard 核心是 "写日志 + 看日志":代码中用 TensorBoard 回调写入日志,终端启动服务查看可视化结果;
  2. 小白入门先聚焦 Scalars 页面,通过损失 / 准确率曲线判断模型训练效果;
  3. 关键代码是 tf.keras.callbacks.TensorBoard(),只需指定日志路径,加到模型训练的 callbacks 中即可。

如果运行中遇到端口被占用(比如提示 6006 被占用),可以指定端口启动:tensorboard --logdir=./tensorboard_logs/ --port=6007,然后访问 http://localhost:6007 即可。

@浙大疏锦行

相关推荐
skywalk8163几秒前
经过多次尝试,在kaggle 双T4训练Qwen2.5-0.5B的正确打开方式是:
人工智能·深度学习·机器学习·qwen2.5·段言
孪生质数-11 分钟前
AI Agent 工程实践(一):大模型 API 接入示范
网络·人工智能·ai·chatgpt·github·claude·claudecode
qq_4195630914 分钟前
机器能思考吗?“——图灵测试、达特茅斯会议与 AI 的诞生
人工智能
维核科技17 分钟前
世界模型落地:物理 AI 方案走向商用,机器人“理解“物理世界
人工智能
神神道呵he20 分钟前
[深度学习] 大模型学习-RAG技术全景解析
人工智能·深度学习·学习
deepseek2324 分钟前
750 亿参数只激活 37 亿:LG 开源 K-EXAONE 2.0,与 DeepSeek 的路线之争迎来新玩家
人工智能·ai agent·mcp
AI街潜水的八角28 分钟前
菌落微生物检测和识别2:基于深度学习YOLO26神经网络实现菌落微生物检测和识别(含训练代码和数据集)
人工智能·深度学习·神经网络
weixin_4462608533 分钟前
ORCA-bench:大模型智能体能否胜任线上故障值守?
人工智能
fthux39 分钟前
装闭 RenoPit 源码解析(13):生成AI装修闭坑PDF报告
人工智能·ai·pdf·开源·github
独行侠影a1 小时前
Mojo:专为AI而生的“Python++”,能否真正挑战CUDA与C++的统治地位?
大数据·人工智能·深度学习