yolov5关闭wandb

2023-05-16

        yolov5训练过程中wandb总是提示登入账号,不登入还不能继续训练,想要关闭wandb,直接不使用即可。

在\yolov5\utils\loggers\wandb\wandb_utils.py中

import wandb的下面加上wandb = None

#try:
#    import wandb
#    from wandb import init, finish
#except ImportError:
#    wandb = None

# 注释掉以上代码,添加以下语句
wandb = None

同时在\yolov5\utils\loggers\__init__.py中

import wandb的位置下面同样加上wandb = None

#try:
#    import wandb
# 
#    assert hasattr(wandb, '__version__')  # verify package import not local dir
#    if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]:
#        wandb_login_success = wandb.login(timeout=30)
#        if not wandb_login_success:
#            wandb = None
#except (ImportError, AssertionError):
#    wandb = None

# 注释掉以上代码,添加以下语句
wandb = None

这样就可以关闭wandb
————————————————
参考:https://blog.csdn.net/athrunsunny/article/details/123428667

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

yolov5关闭wandb 的相关文章

随机推荐