【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"

(待续)

相关推荐
z千鑫6 分钟前
【人工智能】深入理解PyTorch:从0开始完整教程!全文注解
人工智能·pytorch·python·gpt·深度学习·ai编程
爱喝热水的呀哈喽20 分钟前
torch张量与函数表达式写法
人工智能·pytorch·深度学习
MessiGo30 分钟前
Python 爬虫 (1)基础 | 基础操作
开发语言·python
肥猪猪爸1 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
LZXCyrus1 小时前
【杂记】vLLM如何指定GPU单卡/多卡离线推理
人工智能·经验分享·python·深度学习·语言模型·llm·vllm
Enougme1 小时前
Appium常用的使用方法(一)
python·appium
懷淰メ2 小时前
PyQt飞机大战游戏(附下载地址)
开发语言·python·qt·游戏·pyqt·游戏开发·pyqt5
hummhumm2 小时前
第 22 章 - Go语言 测试与基准测试
java·大数据·开发语言·前端·python·golang·log4j
hummhumm2 小时前
第 28 章 - Go语言 Web 开发入门
java·开发语言·前端·python·sql·golang·前端框架
每天吃饭的羊2 小时前
python里的数据结构
开发语言·python