本地浏览器查看云服务器训练模型的tensorboard界面

最近还是在搞YOLO弄毕业设计,一开始是在自己电脑上训练的,但是当开始训练时电脑就卡得不行,别的啥都干不了;另一方面训练速度也特别慢,还得一直开着机不能中断。

索性尝试把模型部署到阿里云服务器上训练,用的最低的2核2G的配置,勉勉强强能跑起来,虽然速度慢了点,甚至完全是用cpu跑的,但好处自然是不占用自己电脑的时间。

这时候就遇到了一个问题,以往部署到自己电脑的时候,可以输入localhost:6006查看tensorboard界面,判断训练情况,但是部署到云服务器上如何查看?

我是用screen指令把YOLOv5的训练挂起来的,使用 screen 后台运行 YOLOv5 训练任务的好处是,即使你断开 SSH 连接,训练任务仍然会继续在后台运行。而在另一个连接中启动 TensorBoard 则可以让你随时监控训练的进度和结果。

新建一个SSH连接

cd到YOLOv5的项目文件夹下,运行tensorboard --logdir runs/train

本地主机进入cmd终端,输入ssh -L 6006:localhost:6006 root@123.123.123.123(“123.123.123.123”替换成云服务器的公网IP)

这个操作的意思是通过SSH隧道将云服务器的 6006 端口转发到本地计算机的 6006 端口

这样,在本地浏览器打开localhost:6006,可以直接查看云服务器的tensorboard面板。

posted @ 2024-05-18 00:21  拾一贰叁  阅读(164)  评论(1编辑  收藏  举报