这个错误通常是由于网络连接出现问题造成的。解决方法是增加连接的超时时间或将W&B的主机地址更改为可靠的地址,例如,更改为Google的地址:
import wandb
wandb.init(project="my-project", entity="my-entity", config={"learning_rate": 0.01})
# 更改W&B主机地址为Google的地址
wandb.login(key='API-key', host='https://wandb.ep.google.com')
# 增加连接的超时时间
os.environ['WANDB_REQUEST_TIMEOUT_S'] = '30'
# 运行W&B超参数搜索
tuner = wandb.sdk.integration.hyperparameter_search.tune(
train_fn=my_train_func,
params=hp,
project="my-project",
entity="my-entity",
objective="validation_loss",
maximize=False,
)