yolov5关闭wandb

yolov5训练过程中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

猜你喜欢

转载自blog.csdn.net/athrunsunny/article/details/123428667