SummaryWriter介绍:
add_scalar()介绍
一、安装TensorBoard
在pytorch环境下安装tensorboard:
pip install tensorboard
安装完毕后 运行代码
一、add_scalar()的使用
- TensorBoard 是一个可视化工具,可以帮助您跟踪和可视化训练过程中的指标、损失值、模型图等信息。
SummaryWriter
类允许您将这些信息写入到 TensorBoard 日志文件中,以便在 TensorBoard 中进行查看和分析。
from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter("logs