windows下Pytorch安装和使用tensorboardX

windows下Pytorch安装和使用tensorboardX

1. 安装

​ pip install tensorboardX

2. 使用

​ 运行如下实例代码

import math
from tensorboardX import SummaryWriter

if __name__ == "__main__":
    writer = SummaryWriter()

    funcs = {"sin": math.sin, "cos": math.cos, "tan": math.tan}

    for angle in range(-360, 360):
        angle_rad = angle * math.pi / 180
        for name, fun in funcs.items():
            val = fun(angle_rad)
            writer.add_scalar(name, val, angle)

    writer.close()

​ 会在当前目录生成runs文件夹

3. 使用

​ cmd下激活你安装tensorboardX的虚拟环境，输入tensorboard --logdir=路径，路径精确到runs文件夹

​

​ 如：

然后会生成一个网址，复制到浏览器即可