添加以下环境
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('./runs')然后对应需要可视化的指标
结束后关闭
训练完成后
会在指定目录下有这个log文件
然后使用下列代码可视化即可
tensorboard --logdir=/home/test/IROS24/pointnet_pytorch/utils/runs/
本文介绍了如何在PyTorch项目中使用TensorBoard的SummaryWriter来记录训练指标，以便在训练完成后在指定目录生成log文件进行可视化。只需创建SummaryWriter对象，添加需要的指标，训练结束后关闭writer，通过命令行运行tensorboard即可查看结果。
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
