一. kaggle_secrets登录wandb
- 在kaggle notebook中添加user_secrets
如下图,选择secrets - 添加secrets,其中value就是你wandb账号的API key
- wandb 账号apikey所在位置
- kaggle中登录代码如下:
import wandb
from kaggle_secrets import UserSecretsClient
user_secrets = UserSecretsClient()
wandb_api = user_secrets.get_secret("wandb_key")
wandb.login(key=wandb_api)
notes = f'steeldefectdetection:{path}'
wandb.init(project="SteelDefectDec_smp_0428",
notes=notes,entity="tkyjqh")
二. 解决wandb.finish出错,不显示jupyter日志问题
如果在代码中增加关闭单元格输出功能将导致wandb中不显示jupyter日志,而且wandb.finish()报错。
%%capture
import wandb
from kaggle_secrets import UserSecretsClient
上面代码在wandb.finish()时报出如下错误:
'ZMQDisplayPublisher' object has no attribute '_orig_publish'
发现这个问题不容易!解决很简单!不要关闭单元格输出!!!
|