安装

pip install -i https://mirrors.aliyun.com/pypi/simple/ tensorboard

使用案例

新建一个python的文件,写入:

from torch.utils.tensorboard import SummaryWriter
import time
import random

writer = SummaryWriter(log_dir='./' + time.strftime('%y-%m-%d_%H.%M', time.localtime())) # 确定路径
for i in range(10):
    # 直接写入就行,更多用法请参考后面的链接,这里就是最最最简单的例子
    writer.add_scalar('test_value', random.random(), i)
# 想写的都写完后调用这个函数
writer.close()

然后进入到目录中,运行ls可以看到创建的目录21-07-18_10.41(这个是与大家自己的时间相同哈)即可,然后运行:

tensorboard --logdir=21-07-18_10.41

得到如下界面:
在这里插入图片描述

参考文章

https://pytorch.org/docs/stable/tensorboard.html

Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐