问题
在使用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)食用风味更佳