安装Tensorboard

执行如下命令安装tensorboard:

pip install tensorboard

在这里插入图片描述
运行如下程序:

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")
# writer.add_image()
for i in range(100):
    writer.add_scalar("y = x", i, i)
writer.close()

即可得到如下事件:

在这里插入图片描述

在PyCharm中的Terminal窗口下执行如下命令打开事件:

tensorboard --logdir=logs #logdir=事件文件所在文件夹名称

点击链接即可得到如下:在这里插入图片描述

还可以使用如下命令指定端口:

tensorboard --logdir=logs --port=6007
Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐