系统相关
首页 > 系统相关> > 【Python】Ubuntu+pytorch下启动tensorboard

【Python】Ubuntu+pytorch下启动tensorboard

作者:互联网

在pytorch代码中加入tensorboard语句

with SummaryWriter(comment='-XXX') as w:
     w.add_graph(model,(input_z,input_x))
 # model有两个输入input_x和input_z 
 # 此时会在当前位置下创建一个runs文件夹
 # 文件夹中会储存当前运行下生成的日志文件

若想要指定日志文件的输出位置:

with SummaryWriter('logs/',comment='-XXX') as w:
     w.add_graph(model,(input_z,input_x))
 # 此时会在当前位置下创建一个logs文件夹
 # 也可以将‘logs/’替换为任意路径位置

默认的runs文件夹和指定的logs文件夹中生成的日志文件储存形式略有不同,前者会在日志文件下额外生成一个带有时间、用户名和自定义标签的文件夹。
在这里插入图片描述

启动tensorboard

以默认的runs文件夹为例,在终端中将路径切换到runs下:

../code/runs$
../code/runs$ tensorboard --logdir=Sep11_23-08-40_xxx-Ubuntu-XXX
# --logdir中的地址为日志文件外的那个文件夹
# 若是logs文件夹的话
# 将地址切换到上层路径
# ../code$ tensorboard --logdir=logs

标签:runs,logs,Python,pytorch,文件夹,tensorboard,input,日志
来源: https://blog.csdn.net/ZZJ__/article/details/100878595