文件名:【Tensorboard+Pytorch】使用注意事项,inspiron one
【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 SummaryWriterwriter
.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”
(待续)