Tensorboard 基础与使用

Tensorboard是TensorFlow中提供的可视化工具，它能可视化数据曲线、模型拓扑图、图像、统计分布曲线等。Pytorch现在也可以使用。

tensorboard安装：

pip install tensorboard

    import os
    from torch.utils.tensorboard import SummaryWriter


    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
    log_dir = BASE_DIR  # 即test_tensorboard.py文件所在目录

    writer = SummaryWriter(log_dir=log_dir, filename_suffix="_test_tensorboard")  # 实例化
   # writer = SummaryWriter(log_dir="logs")
    # writer = SummaryWriter(comment="test01", filename_suffix="_test_tensorboard")
    x = range(100)    # 创建数据
    for i in x:
        writer.add_scalar('y=2x', i * 2, i)
        writer.add_scalar('y=pow(2, x)', 2 ** i, i)
    writer.close()    # 关闭

获得event file文件,terminal运行

tensorboard --logdir=your path dir

运行点击链接