mac如何使用tensorboardx?

1 安装 TensorBoardX

TensorBoardX 是一个用于将 PyTorch 的训练数据可视化到 TensorBoard 的工具库。安装时可以直接使用 pip 命令：

pip install tensorboardX

2 基本使用方法

from tensorboardX import SummaryWriter

writer = SummaryWriter("logs")

for epoch in range(100):
    writer.add_scalar("epoch", epoch, global_step=epoch)
    writer.add_scalar("learning_rate", epoch ** 2, global_step=epoch)

writer.close()

执行之后，会生成日志文件。

3 如何查看日志

在mac上，你还要安装一下

pip3 install tensorboard

其实咱们已经安装了3个tensorboard插件。

查看日志文件：

在浏览器中的查看：