pytorch使用tensorboard

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)

参考文章

• https://blog.csdn.net/bigbennyguo/article/details/87956434
• https://blog.csdn.net/duanlianvip/article/details/98498826