tensorboard 使用方法

  1. 安装tensorboardX

  2. 在训练文件里面 from tensorboardX import SummaryWriter

  3. 插入需要的地方
    在这里插入图片描述
    log_dir里面配置对应的存储文件夹,flush_secs是每隔多少秒保存一次
    这样就把网络结构存在图里面

  4. 在tensorboard里面看训练和验证的loss
    主要加最后一句即可,三个参数对应loss名称,loss值,横坐标
    在这里插入图片描述

  5. 然后把输出的链接在浏览器里面访问即可

    备注:如果看tensorboard过程中,报错 No module named ‘tensorboard’ ,不是未安装库的原因的话,那么使用如下语句

    python -m tensorboard.main --logdir=tf_logs

另外如果想要在windows/mac下看服务器训练结果,需要建立隧道映射

  1. 在cmd/iterm中输入 ssh -L port:127.0.0.1:port account@server.address port自己设置,这个映射必须留着不能删去,否则无法访问tensorboard
    (代替一般ssh远程登录命令:ssh account@server.address

    输入服务器账号的密码

    如果映射端口冲突直接换个端口

  2. 在服务器中运行 python -m tensorboard.main --logdir=文件地址

    最后windows/mac浏览器访问地址:http://127.0.0.1:6006/

参考:部分内容参考在服务器使用TensorBoard的方法

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值