-
安装tensorboardX
-
在训练文件里面 from tensorboardX import SummaryWriter
-
插入需要的地方
log_dir里面配置对应的存储文件夹,flush_secs是每隔多少秒保存一次
这样就把网络结构存在图里面 -
在tensorboard里面看训练和验证的loss
主要加最后一句即可,三个参数对应loss名称,loss值,横坐标
-
然后把输出的链接在浏览器里面访问即可
备注:如果看tensorboard过程中,报错 No module named ‘tensorboard’ ,不是未安装库的原因的话,那么使用如下语句
python -m tensorboard.main --logdir=tf_logs
另外如果想要在windows/mac下看服务器训练结果,需要建立隧道映射
-
在cmd/iterm中输入
ssh -L port:127.0.0.1:port account@server.address
port自己设置,这个映射必须留着不能删去,否则无法访问tensorboard
(代替一般ssh远程登录命令:ssh account@server.address
)输入服务器账号的密码
如果映射端口冲突直接换个端口
-
在服务器中运行
python -m tensorboard.main --logdir=文件地址
最后windows/mac浏览器访问地址:http://127.0.0.1:6006/
参考:部分内容参考在服务器使用TensorBoard的方法