文章目录
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