1、TensorBoard介绍
TensorBoard是TensorFlow的可视化工具,但它也可以与PyTorch结合使用。TensorBoard提供了一个Web界面,可以展示你训练过程中的各种信息,如损失值、准确度、权重分布等,更好地帮助开发者理解和调试模型。
TensorBoard | TensorFlow (google.cn)https://tensorflow.google.cn/tensorboard?hl=zh-cn
2、导入相关包
安装Tensor包
pip install tensorboard
导入TensorBoard相关包,要导入SummaryWriter模块
import torch
from torch.utils.tensorboard import SummaryWriter