pip install tensorboard
from torch.utils.tensorboard import SummaryWriter
loss_tensor_board = SummaryWriter(path)
loss_tensor_board.add_scalar('loss', loss, 第几个epcoh)
cmd 打开命令行 切换到conda虚拟环境(安装的有tensorboard)
直接在命令行中运行python -m tensorboard.main --logdir=path(这个path就是loss_tensor_board = SummaryWriter(path) 这里路径)

将http://localhost:6006/ 复制到浏览器打开即可

1349

被折叠的 条评论
为什么被折叠?



