前言
在训练模型会生成
events.out.tfevents.***.**
代码段,可以通过Tensorboard可视化查看训练过程中的损失函数变化或者模型的结构。
所需环境
conda create -n Berber python=3.6 #创建名为Berber的环境
conda activate Berber # 激活环境Berber
tensorflow
pip install tensorflow-gpu
tensorboard
pip install tensorboard -i https://pypi.douban.com/simple
注意tf和tensorbord版本匹配,我用的版本
##python=3.7
pip install tensorflow-gpu==2.1.0
pip install tensorboard==2.1.1
Tensorboard使用
任意打开一个python文件,配置好环境,打开终端
输入
tensorboard --logdir=D:\Berber\events --port=8008
D:\Berber\events 是events.out.tfevents文件所在的文件目录
--port=8008 是开启可视化的“地址”,修改 --port= 后面的数字可以同时可视化多个不同的events.out.tfevents文件
终端输入后,将其复制到浏览器即可
参考:tensorboard可视化(可视化查看events.out.tfevents文件) - 知乎 (zhihu.com)