使用tensorboard实时查看训练效果

365游戏厅平台 2025-08-15 14:44:08 admin

一、安装tensorboard

tensorboard原属于tensorflow模块组件,后pytorch接入了tensorboard的api,只需在pytorch的环境中执行以下命令。

pip install tensorboard

安装的过程可能会有些缓慢,耐心等待或者更换国内镜像源进行下载。

二、运行tensorboard

命令行输入

tensorboard --logdir=相对路径

这个相对路径指的是你训练生成的文件的路径。这里以yolov8为例,相对路径就是runs/detect/train6。

须知,使用tensorboard查看实时训练效果需要文件下包含下面这种文件。一般,安装tensorboard之后,再训练模型时会自动产生。

三、实时查看训练情况

命令行输入命令运行之后,会返回一个网址,点击打开网址,会出现下面图片的网站。

如果保持默认,网站不会自动刷新,需要修改。点击右上角的设置,也就是齿轮按钮,

勾选Reload data。关闭之后,网站就会自动刷新了。

相关文章

人口,女性(占总人口的百分比)

3月21日体验服停机更新公告|【暃】技能重新调整优化,终极觉醒新增瑶等英雄!

老鼠活动的时间规律是怎么样?