【Tensorboard+Pytorch】使用注意事项

安装

  • tensorboard/tensorboardx版本需要与tensorflow保持一致(本人使用2.2)

调用

  • 环境变量
    在终端或CMD中使用时,常见报错"tensorboard 不是内部或外部命令......",需要添加环境变量路径path。具体为tensorboard.exe所在目录(Anaconda/Scripts)和(Anaconda/Library/bin)
  • 命令
python 复制代码
 tensorboard --logdir=./scalar --port=6006

【注意】logdir的路径层级关系,层级不对,找不到event文件,则无法显示

代码结构

python 复制代码
from tensorboardX import SummaryWriter

writer.add_scalar('scalar/eva_reward', eva_reward, evaluate_num)
writer.close()

add_scalar 三个参数分别是名称、y轴变量、x轴变量

【注意】要close,不然会报错"tensorflow:Unable to get first event timestamp for run"

(待续)

相关推荐
Luchang-Li2 分钟前
sglang pytorch NCCL hang分析
pytorch·python·nccl
一个天蝎座 白勺 程序猿5 小时前
Python爬虫(47)Python异步爬虫与K8S弹性伸缩:构建百万级并发数据采集引擎
爬虫·python·kubernetes
XiaoMu_0016 小时前
基于Django+Vue3+YOLO的智能气象检测系统
python·yolo·django
Gyoku Mint7 小时前
提示词工程(Prompt Engineering)的崛起——为什么“会写Prompt”成了新技能?
人工智能·pytorch·深度学习·神经网络·语言模型·自然语言处理·nlp
honder试试7 小时前
焊接自动化测试平台图像处理分析-模型训练推理
开发语言·python
心本无晴.7 小时前
Python进程,线程
python·进程
java1234_小锋10 小时前
Scikit-learn Python机器学习 - 特征降维 压缩数据 - 特征提取 - 主成分分析 (PCA)
python·机器学习·scikit-learn
java1234_小锋10 小时前
Scikit-learn Python机器学习 - 特征降维 压缩数据 - 特征提取 - 线性判别分析 (LDA)
python·机器学习·scikit-learn
思辨共悟11 小时前
Python的价值:突出在数据分析与挖掘
python·数据分析
计算机毕业设计木哥11 小时前
计算机毕设选题:基于Python+Django的B站数据分析系统的设计与实现【源码+文档+调试】
java·开发语言·后端·python·spark·django·课程设计