1安装
pip install tensorboardX 本地测试使用版本为tensorboard=1.8 pytorch=0.4.1
from tensorboardX import SummaryWriter
# Creates writer1 object.
# The log will be saved in 'runs/exp'
writer1 = SummaryWriter('runs/exp')
# Creates writer2 object with auto generated file name
# The log directory will be something like 'runs/Aug20-17-20-33'
writer2 = SummaryWriter()
# Creates writer3 object with auto generated file name, the comment will be appended to the filename.
# The log directory will be something like 'runs/Aug20-17-20-33-resnet'
writer3 = SummaryWriter(comment='resnet')
2查看log
当使用–bind_all时候,可以忽略公司配置代理,在本机电脑直接显示。
tensorboard --logdir=<your_log_dir> --bind_all
3详细使用tensorboard
-
add_scalar(tag, scalar_value, global_step=None, walltime=None)
-
add_image(tag, img_tensor, global_step=None, walltime=None, dataformats=‘CHW’)
-
add_histogram(tag, values, global_step=None, bins=‘tensorflow’, walltime=None, max_bins=None)
-
add_graph(model, input_to_model=None, verbose=False, **kwargs)
-
add_embedding(mat, metadata=None, label_img=None, global_step=None, tag=‘default’, metadata_header=None)
参考文章