欢迎光临散文网 会员登陆 & 注册

yolov5如何关闭wandb

2021-09-09 11:18 作者:ZhouBay  | 我要投稿

    由于yolo训练过程中还是最后输出的结果信息都是比较全面的,所以有时我们不需要这个最强可视化工具。

  • 修改:

    yolo项目目录下的 utils/wandb_logging/wandb_utils.py,可以直接在处理异常的代码外将wandb = None,将wandb扼杀在try的摇篮外。

  • 训练:

    这时只会输出一个警告提示(让你安装wandb),但是不影响训练。

     直接在它加载库的地方去修改是最方便的也是最有效的方法,但是也有头铁的去改别的地方的。

yolov5如何关闭wandb的评论 (共 条)

分享到微博请遵守国家法律