这里写自定义目录标题
Pytorch TensorboardX可视化
安装
pip install tensorboardX
pip install tensorboard
代码嵌入
from torch.utils.tensorboard import SummaryWriter
w = SummaryWriter(‘runs/’)
w.add_scalar('loss", loss, epoch)
本地查看服务器日志可视化
本地输入ssh进入服务器
ssh -L 6006:127.0.0.1:6006 name@10.20.28.74
运行代码并生成链接
tensorboard --logdir=runs/ --host=127.0.0.1
本地浏览器打开查看可视化
http://127.0.0.1:6006