Pytorch 配置 Tensorboard 可视化学习(一)
安装pip install -i https://mirrors.aliyun.com/pypi/simple/ tensorboard使用案例新建一个python的文件,写入:from torch.utils.tensorboard import SummaryWriterimport timeimport randomwriter = SummaryWriter(log_dir='./' +
·
安装
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

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