YOLOv5-7.0解决报错 wandb: Network error (TransientError), entering retry loop.

前言

最近在复习yolov5目标检测代码时用了yolov5的最新7.0版本,之前用的是5.0版本,这一新版本相对于之前做了一些提升,对于package的兼容也要好了很多,但也不是说下载了直接就能运行,实际使用过程中还是遇到了许多新的问题,下面就我自己碰到的问题提出解决方法。

问题

wandb是非常好用的可视化工具,但是国内的话,使用时常常会无法同步数据,需要借助魔法来连接服务器,而yolov5的源码恰恰是用了wandb来可视化训练结果,所以在训练时,如果我们的网络不好,就会出现如下所示的问题:

可以看出一直在重连网络,此时我的训练结果已经出来了,但是程序仍在运行中,这对我们后续使用pycharm或者其他工具都很不好。

解决

这里提供一种最简单的方法,那就是暂时取消同步wandb,等网络配置好之后再同步上去。

在train.py的前面几行(加在import导入模块后面就行),添加如下的代码:

os.environ["WANDB_API_KEY"] = YOUR_KEY_HERE
os.environ["WANDB_MODE"] = "offline"

其中,YOUR_KEY_HERE这里要改成你的wandb的key,在这个网址可以找到https://wandb.ai/settings

前提是你要注册一个账号,很简单的,具体在这里找key,如下所示:

复制完key之后,添加代码到train.py即可,具体如下所示:然后再运行train.py就不会一直重连服务器了,并且与运行窗口会给出相应的措施,如果想看可视化结果的话,就等自己的网络好了,或者换一台可以使用的设备,按照它给出的措施进行同步就可以了。

结尾

菜鸟一只,希望这篇文章可以帮助大家,有什么不足请各位大佬批评指正,喜欢的话点个关注哦,后续还会更新yolov5相关内容。

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值