文章目录
一、TensorBoard简介
机器学习可视化工具,用来记录机器学习过程。
TensorBoard界面展示:
二、TensorBoard安装与运行
#安装命令
pip install tensorboard
安装后再pycharm终端输入
tensorboard --help
若能正常显示,则说明运行成功。
三、Pytorch使用TensorBoard
Pytorch使用Tensorboard主要用到了三个API:
. SummaryWriter:这个用来创建一个log文件,TensorBoard面板查看时,也是需要选择查看那个log文件。
. add_something: 向log文件里面增添数据。例如可以通过add_scalar增添折线图数据,add_image可以增添图片。
. close:当训练结束后,我们可以通过close方法结束log写入。
四、总结
总的来说,pytorch使用TensorBoard的过程就是:使用SummaryWriter()创建一个log文件,然后使用add_something()或者add_image()往log文件里面添加数据,之后在tensorboard中看到的数据来源于这个log文件。