【pytorch】在python脚本中自动运行tensorboard

问题

在使用tensorboard进行训练过程可视化时，每开始一次新的实验，就需要在终端重新运行一次tensorboard --logdir=path_to_log，这比较麻烦，尤其是在代码调试阶段，我们会频繁地开始新的实验，

那么如何在python脚本中自动打开tensorboard呢？

解决方法

原代码

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir=path_to_logs) #日志存储的路径

修改后的代码

from torch.utils.tensorboard import SummaryWriter
from tensorboard import program


writer = SummaryWriter(log_dir=path_to_logs) #日志存储的路径

# 新增代码
tb = program.TensorBoard()
tb.configure(argv=[None, '--logdir', path_to__logs])
url = tb.launch()

在浏览器中一直开着localhost:6006，如果重新开始新的实验，只需要刷新浏览器就可以查看最新实验的训练过程了。

配合查看远端服务器上的Tensorboard（利用MobaXterm）食用风味更佳

参考资料

1. How to run Tensorboard from python scipt in virtualenv?