- 首先在环境中安装相关的库
pip install tensorboardX # pip install 'git+https://github.com/lanpa/tensorboardX'
下面另两个库可以选择性安装
pip install crc32c
pip install soundfile
- 代码
# 导入相关库
from tensorboardX import SummaryWriter
# 创建实例
writer = SummaryWriter(log_dir="文件夹名称", comment='相当于备注') # log_dir默认为runs comment默认是none
# 保存相关数据
writer.add_scaler(‘名称’, Y轴数据, X轴数据)
#结束
writer.close()
可视化命令语句
tensorboard命令行
(1)进入程序所在的虚拟环境:conda activate XXX
(2)tensorboard --logdir=……\log
GitHub - lanpa/tensorboardX: tensorboard for pytorch (and chainer, mxnet, numpy, ...)