PyTorch 使用 tensorboard

PyTorch 使用 tensorboard

1. 安装 tensorboardx 和 tensorboard
   
2. 在训练中通过 tensorboardx.SummaryWriter 记录数据， 例如：

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()

4. 找到环境对应的 CMD.exe 并打开
   
5. 输入运行文件所在的盘符，如 F:
6. 输入 cd + runs文件夹的上一层文件的路径，例如：cd \tensorboard_test
   
7. 输入 tensorboard --logdir runs
   
8. 复制 http://localhost:6006/ 到浏览器中打开