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/87956434https://blog.csdn.net/duanlianvip/article/details/98498826