wandb关闭问题

最近在训练模型的时候,总是弹出让我登录wandb官网,先暂且不谈使用wandb的好处。

第一次就把wandb注册了,记得好像需要挂VPN才可以访问。

但是后来才发现,每一次进行训练都会出现wandb: Currently logged in as: liudawei. Use `wandb login --relogin` to force relogin.这就意味着每次都得登录一下,虽说登录可以看到训练的状态,是个非常好的工具,但每次都登录有点浪费时间了。

其他博客说明 这么关闭wandb

在网上找了好多好多篇博客,几乎每一片都是说明在两处代码的地方加上wandb = None

第一处:utils/loggers/wandb/wandb_utils.py

try:
    import wandb

    assert hasattr(wandb, '__version__')  # verify package import not local dir
except (ImportError, AssertionError):
    wandb = None
# 添加以下语句
wandb = None

第二处:utils/loggers/ __init __.py

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}:
        try:
            wandb_login_success = wandb.login(timeout=30)
        except wandb.errors.UsageError:  # known non-TTY terminal issue
            wandb_login_success = False
        if not wandb_login_success:
            wandb = None
except (ImportError, AssertionError):
    wandb = None
# 添加以下语句
wandb = None

然而我发现并没有解决问题。

其实是这么关闭wandb

第一处:utils/loggers/wandb/wandb_utils.py

# try:
#     import wandb
# 
#     assert hasattr(wandb, '__version__')  # verify package import not local dir
# except (ImportError, AssertionError):
#     wandb = None
# 添加以下语句
wandb = None

第二处:utils/loggers/ __init __.py

# 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}:
#         try:
#             wandb_login_success = wandb.login(timeout=30)
#         except wandb.errors.UsageError:  # known non-TTY terminal issue
#             wandb_login_success = False
#         if not wandb_login_success:
#             wandb = None
# except (ImportError, AssertionError):
#     wandb = None
# 添加以下语句
wandb = None
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐