1.TensorBoard的安装
1.1 关于TensorBoard
TensorBoard是Google开发的一组用于数据可视化的工具。它包含在流行的开源机器学习库 Tensorflow 中,主要用于记录机器学习的过程。
- 跟踪和可视化损失及准确率等指标
- 可视化模型图(操作和层)
- 查看权重、偏差或其他张量随时间变化的直方图
- 将嵌入投射到较低的维度空间
- 显示图片、文字和音频数据
- 剖析 TensorFlow 程序
【嘿,没看懂。】
另篇文章介绍的功能是:
- 记录损失变化、准确率变化等
- 记录图片变化、语音变化、文本变化等,例如在做GAN时,可以过一段时间记录一张生成的图片
- 绘制模型
【这个好理解了点。】
1.2 TensorBoard的安装(只看加粗字体就行)
在命令行中,先进入虚拟环境:activate pytorch
然后使用pip进行安装:pip install tensorboard
然后运行示例代码:
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
for i in range(100):
writer.add_scalar("y=x", i, i)
writer.close()
发现报错,错误原因是TensorBoard版本过低:
在终端中输入conda list 查看Tensorboard版本,1.10.0,确实低。