【Tensorboard+Pytorch】使用注意事项

安装

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

调用

• 环境变量
  在终端或CMD中使用时，常见报错"tensorboard 不是内部或外部命令......"，需要添加环境变量路径path。具体为tensorboard.exe所在目录(Anaconda/Scripts)和(Anaconda/Library/bin)
• 命令

 tensorboard --logdir=./scalar --port=6006

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

代码结构

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"

（待续）